Update app.py

app.py

@@ -1,10 +1,31 @@
 from transformers import pipeline
 import gradio as gr
 
-model = pipeline("text-generation", model="ButteredToast/ToastBot")
+# Load model safely
+try:
+    model = pipeline("text-generation", model="ButteredToast/ToastBot")
+except Exception as e:
+    print(f"Error loading model: {e}")
+    model = None  # Prevent crashes
 
+# Define text generation function
 def generate_text(input_text):
-    return model(input_text)[0]["generated_text"]
+    if model is None:
+        return "Error: Model could not be loaded. Check your Hugging Face space logs."
+    
+    try:
+        output = model(input_text, max_length=100)  # Control output length
+        return output[0]["generated_text"]
+    except Exception as e:
+        return f"Error generating text: {e}"
+
+# Set up Gradio interface
+demo = gr.Interface(
+    fn=generate_text, 
+    inputs="text", 
+    outputs="text",
+    title="ToastBot AI",
+    description="Chat with ToastBot, your AI assistant!"
+)
 
-demo = gr.Interface(fn=generate_text, inputs="text", outputs="text")
 demo.launch()