Update app.py

app.py

@@ -4,9 +4,9 @@ import numpy as np
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-# Load the Qwen model and tokenizer
-tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-0.5B")
-model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2-0.5B")
+# Load the TinyLlama model and tokenizer
+tokenizer = AutoTokenizer.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
+model = AutoModelForCausalLM.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
 
 class Agent:
     def __init__(self, id, api_key=None):
@@ -125,4 +125,4 @@ iface = gr.Interface(
     """
 )
 
-iface.launch()
+iface.launch()