Update app.py

app.py

@@ -3,7 +3,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 import spaces
 
-model_name = "MBZUAI-Paris/Atlas-Chat-27B"
+model_name = "MBZUAI-Paris/Atlas-Chat-9B"
 dtype = torch.bfloat16
 model = AutoModelForCausalLM.from_pretrained(
     model_name, 
@@ -25,25 +25,33 @@ def chat(input_text, history=[]):
     return history, history
 
 
-iface = gr.Interface(
-    fn=chat,
-    inputs=[
-        gr.Textbox(label="أدخل رسالتك هنا"), 
-       gr.State()
-    ],
-    outputs=[
-        gr.Chatbot(label="المحادثة"),  
-        gr.State()
-    ],
-    live=True,
-    title="دردشة أطلس",
-    description="تطبيق دردشة يعمل بنموذج أطلس-شات لتوفير تفاعل ذكي وسلس",
-    theme="huggingface",
-    examples=[
-        ["مرحباً! كيف يمكنني مساعدتك اليوم؟"],
-        ["ما هي أخبار التكنولوجيا الحديثة؟"]
-    ]
-)
+with gr.Blocks() as app:
+    gr.Markdown("<h1 style='text-align: center;'>دردشة أطلس</h1>")
+    chatbot = gr.Chatbot(label="المحادثة")
+    state = gr.State([])
+
+    with gr.Row():
+        txt = gr.Textbox(show_label=False, placeholder="اكتب رسالتك هنا...").style(container=False)
+        send_button = gr.Button("إرسال")
+
+    # Define the button click event
+    def user_input(input_text, history):
+        return "", history + [[input_text, None]]
+
+    def bot_response(history):
+        user_message = history[-1][0]
+        inputs = tokenizer(user_message, return_tensors="pt").to(model.device)
+        outputs = model.generate(**inputs, max_new_tokens=150)
+        response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+        history[-1][1] = response
+        return history
+
+    # Link functions to button clicks
+    txt.submit(user_input, [txt, state], [txt, state]).then(
+        bot_response, state, chatbot
+    )
+    send_button.click(user_input, [txt, state], [txt, state]).then(
+        bot_response, state, chatbot
+    )
 
-# Launch the application
-iface.launch()
+app.launch()