Update app.py

app.py

@@ -2,29 +2,29 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-MODEL_NAME = "Qwen/Qwen2.5-7B-Instruct"
+MODEL_NAME = "Qwen/Qwen1.5-4B"
 
 # بارگذاری مدل و توکنایزر
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_NAME,
-    torch_dtype=torch.float32,  # از float32 استفاده کن تا روی CPU اجرا بشه
+    torch_dtype=torch.float32  # روی CPU اجرا می‌شه
 )
 
 # تابع چت‌بات
 def chat_with_qwen(prompt):
     inputs = tokenizer(prompt, return_tensors="pt")
-    output = model.generate(**inputs, max_new_tokens=200)
+    output = model.generate(**inputs, max_new_tokens=100)
     response = tokenizer.decode(output[0], skip_special_tokens=True)
     return response
 
-# ایجاد رابط کاربری با Gradio
+# رابط Gradio
 iface = gr.Interface(
     fn=chat_with_qwen,
     inputs=gr.Textbox(lines=2, placeholder="سوال خود را اینجا بنویسید..."),
     outputs="text",
-    title="Qwen 2.5 7B Chatbot",
-    description="چت‌بات مبتنی بر مدل Qwen2.5-7B-Instruct",
+    title="Qwen 1.5 4B Chatbot",
+    description="چت‌بات با مدل سبک‌تر برای منابع رایگان",
 )
 
 iface.launch()