Update app.py

app.py

@@ -32,7 +32,7 @@ def generate_text(prompt, model_name):
 
 # Create Gradio interface
 demo = gr.Interface(
-    fn=generate_story,
+    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...")