Update app.py

app.py

@@ -7,14 +7,15 @@ generator = InferenceClient("mistralai/Codestral-22B-v0.1")
 # Define the function for text generation
 def 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
-        )
+        # Generate text using the correct API call
+        result = generator.text_generation(text, details=True, stream=False)
         # Extract and return the generated text
-        return result["generated_text"] if "generated_text" in result else "Error: No text generated."
+        if "generated_text" in result:
+            return result["generated_text"]
+        elif isinstance(result, list) and len(result) > 0:
+            return result[0]  # If the response is a list, return the first item
+        else:
+            return "Error: Unable to generate text."
     except Exception as e:
         return f"Error: {e}"