Update app.py

app.py

@@ -1,13 +1,19 @@
 import gradio as gr
-from transformers import pipeline
+from huggingface_hub import InferenceClient
 
 # Load the text generation pipeline
-generator = pipeline('text-generation', model='gpt2')
+generator = InferenceClient("mistralai/Codestral-22B-v0.1")
 
 # Define the function for text generation
 def generate(text):
-    result = generator(text, max_length=100, num_return_sequences=1)
-    return result[0]['generated_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"
 
 # Define examples
 examples = [
@@ -26,4 +32,5 @@ run = gr.Interface(
 )
 
 # Launch the app
-run.launch()
+if __name__ == "__main__":
+    run.launch()