Update app.py

app.py

@@ -3,7 +3,6 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 import torch
 import spaces
 
-# Load models for English and Arabic
 def load_model(model_name):
     device = "cuda" if torch.cuda.is_available() else "cpu"
     model = AutoModelForCausalLM.from_pretrained(
@@ -19,12 +18,12 @@ def load_model(model_name):
         tokenizer=tokenizer,
         return_full_text=False,
         max_new_tokens=500,
-        do_sample=True  # Changed to enable sampling for more creative outputs
+        do_sample=False
     )
     return generator
 
 @spaces.GPU
-def generate_text(model_name, prompt):
+def generate_text(prompt, model_name):
     generator = load_model(model_name)
     messages = [{"role": "user", "content": prompt}]
     output = generator(messages)
@@ -34,17 +33,20 @@ def generate_text(model_name, prompt):
 demo = gr.Interface(
     fn=generate_text,
     inputs=[
-        gr.Radio(choices=["microsoft/Phi-3-mini-4k-instruct", "ALLaM-AI/ALLaM-7B-Instruct-preview"], label="Select Language"),
-        gr.Textbox(lines=2, placeholder="Enter your story prompt here...")
+        gr.Textbox(lines=2, placeholder="Enter your prompt here..."),
+        gr.Dropdown(
+            choices=["Qwen/Qwen2.5-1.5B-Instruct","microsoft/Phi-3-mini-4k-instruct", "ALLaM-AI/ALLaM-7B-Instruct-preview"],
+            label="Choose Model",
+            value="ALLaM-AI/ALLaM-7B-Instruct-preview"
+        )
     ],
-    outputs=gr.Textbox(label="Generated Story"),
-    title="Kids Storyteller",
-    description="Choose a language and enter a prompt to generate a fun story for kids!",
+    outputs=gr.Textbox(label="Generated Text"),
+    title="Text Generator",
+    description="Enter a prompt and generate text using one of the available models.",
     examples=[
-        ["microsoft/Phi-3-mini-4k-instruct", "Once upon a time in a magical forest..."],
-        ["ALLaM-AI/ALLaM-7B-Instruct-preview", "في قديم الزمان في غابة سحرية..."]
+        ["Tell me a funny joke about chickens.", "microsoft/Phi-3-mini-4k-instruct"],
+        ["أخبرني نكتة مضحكة عن الدجاج.", "ALLaM-AI/ALLaM-7B-Instruct-preview"]
     ]
 )
 
 demo.launch()
-