Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-MODEL_NAME = "tiiuae/falcon-rw-1b" #"TinyLlama/TinyLlama-1.1B-Chat-v1.0"
+MODEL_NAME = "google/gemma-3-1b-it" #"TinyLlama/TinyLlama-1.1B-Chat-v1.0"
 
 # Load tokenizer and model
 print("Loading tokenizer...")
@@ -69,7 +69,7 @@ demo = gr.Interface(
         gr.Slider(minimum=0, maximum=100, value=50, step=1, label="Top K")
     ],
     outputs=gr.Textbox(label="Generated Text"),
-    title="TinyLlama Gradio API",
+    title="gemma-3-1b-it Gradio API",
     description="Use this via UI or API via `/run/predict`"
 )