Update app.py

app.py

@@ -38,23 +38,24 @@ def generate_text(prompt, max_length=256, temperature=0.7, top_p=0.9, top_k=150,
     )
     return tokenizer.decode(output[0], skip_special_tokens=True)
 
-# Create the Gradio interface
-iface = gr.Interface(
-    fn=generate_text,
-    inputs=[
-        gr.Textbox(label="Prompt: دخل النص بالدارجة"),
-        gr.Slider(50, 500, value=256, label="Max Length"),
-        gr.Slider(0.1, 1.5, value=0.7, label="Temperature"),
-        gr.Slider(0.1, 1.0, value=0.9, label="Top-p"),
-        gr.Slider(1, 10000, value=150, label="Top-k"),
-        gr.Slider(1, 20, value=8, label="Number of Beams"),
-        gr.Slider(0.0, 100.0, value=1.5, label="Repetition Penalty"),
-    ],
-    outputs=gr.Textbox(label="Generated Text in Moroccan Darija"),
-    title="Moroccan Darija LLM",
-    description="Enter a prompt and get AI-generated text using our pretrained LLM on Moroccan Darija.",
-    examples=examples,
-)
-
 if __name__ == "__main__":
-    iface.launch(share=True)
+    # Create the Gradio interface
+    with gr.Blocks() as app:
+        gr.Interface(
+            fn=generate_text,
+            inputs=[
+                gr.Textbox(label="Prompt: دخل النص بالدارجة"),
+                gr.Slider(50, 500, value=256, label="Max Length"),
+                gr.Slider(0.1, 1.5, value=0.7, label="Temperature"),
+                gr.Slider(0.1, 1.0, value=0.9, label="Top-p"),
+                gr.Slider(1, 10000, value=150, label="Top-k"),
+                gr.Slider(1, 20, value=8, label="Number of Beams"),
+                gr.Slider(0.0, 100.0, value=1.5, label="Repetition Penalty"),
+            ],
+            outputs=gr.Textbox(label="Generated Text in Moroccan Darija"),
+            title="Moroccan Darija LLM",
+            description="Enter a prompt and get AI-generated text using our pretrained LLM on Moroccan Darija.",
+            examples=examples,
+        )
+    
+    app.launch(ssr=False)