Update app.py

app.py

@@ -1,39 +1,26 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import pipeline
 
-# Load the Mistral model and tokenizer
-model_name = 'mistral/mistral-7b'  # Replace with your specific Mistral model
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+# Initialize the text generation pipeline with the Mistral model
+pipe = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3")
 
-# Define a function to handle translation
-def translate_text(text, src_lang, tgt_lang):
-    # Prepare the input text for the model
-    # For simplicity, we'll just append source and target languages to the input text
-    input_text = f"{src_lang} to {tgt_lang}: {text}"
-    inputs = tokenizer(input_text, return_tensors="pt")
-    translated = model.generate(**inputs)
-    translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
-    return translated_text
-
-def chat(message, src_lang, tgt_lang):
-    translated_message = translate_text(message, src_lang, tgt_lang)
-    return translated_message
-
-# Define the language options
-languages = ["English", "French", "German", "Spanish", "Chinese"]  # Extend this list as needed
+# Define the function to handle chat
+def chat(message):
+    # Prepare the input format
+    messages = [{"role": "user", "content": message}]
+    # Generate the response using the model
+    response = pipe(messages)
+    # Extract and return the generated text
+    return response[0]['generated_text']
 
 # Create the Gradio interface
 interface = gr.Interface(
     fn=chat,
-    inputs=[
-        gr.inputs.Textbox(label="Enter text"),
-        gr.inputs.Dropdown(choices=languages, label="Source Language"),
-        gr.inputs.Dropdown(choices=languages, label="Target Language")
-    ],
+    inputs=gr.inputs.Textbox(label="Enter your message"),
     outputs="text",
-    title="Mistral Translation Chatbot",
-    description="Translate text between different languages using the Mistral model."
+    title="Mistral Chatbot",
+    description="Chat with the Mistral-7B-Instruct model to get responses to your queries."
 )
 
+# Launch the Gradio interface
 interface.launch()