used transformer instead of openai

app.py

@@ -10,11 +10,13 @@ from transformers import pipeline
 import gradio as gr
 
 # Load the model
-generator = pipeline("text-generation", model="Distilgpt2")
+generator = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1")
 
 # Define your function
 def generate_text(prompt):
-    result = generator(prompt, max_length=100, num_return_sequences=1)
+    result = generator(prompt, max_length=100, num_return_sequences=1, do_sample=True, 
+    temperature=0.7, 
+    top_p=0.9,)
     return result[0]["generated_text"]
 
 # Create Gradio interface