Update app.py

app.py

@@ -3,14 +3,19 @@ from transformers import pipeline
 
 # Load a text-generation model.
 # For demonstration, we're using GPT-2.
-# For an adult-novel style, you may consider a fine-tuned model or adjust parameters.
+# For an adult-novel style, consider a fine-tuned model or adjust parameters.
 generator = pipeline("text-generation", model="gpt2")
 
 def generate_text(prompt, max_length):
-    # Generate text based on the prompt.
-    # Adjust temperature and sampling settings as desired.
-    results = generator(prompt, max_length=max_length, do_sample=True, temperature=0.8)
-    return results[0]["generated_text"]
+    try:
+        # Generate text based on the prompt.
+        results = generator(prompt, max_length=max_length, do_sample=True, temperature=0.8)
+        return results[0]["generated_text"]
+    except Exception as e:
+        # Log the exception details for debugging
+        print("Error during text generation:", e)
+        # Return an error message for the user
+        return "An error occurred while generating text. Please try again with a shorter prompt or lower max length."
 
 # Create the Gradio interface.
 iface = gr.Interface(