Update app.py

app.py

@@ -1,19 +1,22 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# Load the text generation pipeline
+# Load the text generation model
 generator = InferenceClient("mistralai/Codestral-22B-v0.1")
 
 # Define the function for text generation
 def generate(text):
-    # Use the `text_generation` method for generating text
-    result = generator.text_generation(
-        inputs=text,
-        max_length=100,
-        num_return_sequences=1
-    )
-    # Extract and return the generated text
-    return result[0]['generated_text'] if 'generated_text' in result[0] else "Error: Failed to generate text"
+    try:
+        # Use the `text_generation` method correctly
+        result = generator.text_generation(
+            text,  # Pass the text input directly
+            max_length=100,
+            num_return_sequences=1
+        )
+        # Extract and return the generated text
+        return result["generated_text"] if "generated_text" in result else "Error: No text generated."
+    except Exception as e:
+        return f"Error: {e}"
 
 # Define examples
 examples = [