Update app.py

app.py

@@ -8,6 +8,22 @@ def generate_text(prompt, length=50, temperature=0.7, seed=42):
     output = generator(prompt, max_length=length, do_sample=True, temperature=temperature)
     return output[0]['generated_text']
 
+import gradio as gr
+from transformers import pipeline, set_seed
+import logging
+
+logging.basicConfig(level=logging.INFO)
+
+def generate_text(prompt, length=50, temperature=0.7, seed=42):
+    try:
+        set_seed(seed)
+        generator = pipeline('text-generation', model='flax-community/miniLM-L6-h384-uncased', device=0)
+        output = generator(prompt, max_length=length, do_sample=True, temperature=temperature)
+        return output[0]['generated_text']
+    except Exception as e:
+        logging.error(f"Error generating text: {e}")
+        return "Error generating text. Please try again later."
+
 inputs = gr.inputs.Textbox(lines=5, label="Prompt")
 outputs = gr.outputs.Textbox(label="Output Text")
 temperature_slider = gr.inputs.Slider(minimum=0.1, maximum=1.5, default=0.7, label="Temperature")