Update app.py

app.py

@@ -1,13 +1,30 @@
 import streamlit as st
 from transformers import pipeline
 
-# Load the chatbot model
-chatbot = pipeline("conversational")
+# Load the Mistral-7B model using pipeline
+pipe = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.2")
 
-# Streamlit UI
+def generate_response(prompt):
+  """Generates a response using the Mistral-7B model."""
+  response = pipe(prompt, max_length=1000, num_return_sequences=1)[0]["generated_text"]
+  return response.strip()
+
+# Streamlit app layout
 st.title("Mistral Chatbot")
+user_input = st.text_input("Ask me anything!")
 
-user_input = st.text_input("You: ", "")
 if user_input:
-    response = chatbot(user_input)[0]['generated_text']
-    st.text_area("Mistral:", value=response, height=200)
+  response = generate_response(user_input)
+  st.write(f"Mistral: {response}")
+
+# Deployment to Hugging Face Spaces (instructions included)
+# 1. Create a Hugging Face account (if you don't have one)
+# 2. Create a new Space from your account
+# 3. Push your code to a Git repository (e.g., GitHub)
+# 4. In your Space settings, connect your Git repository
+# 5. Under "Model", select the Mistral-7B model you're using
+# 6. Under "Environment", create a new environment with Python 3.7+
+# 7. Under "Requirements", add "streamlit transformers" (separate lines)
+# 8. Under "Start script", enter "streamlit run app.py" (replace app.py with your filename)
+# 9. Deploy your Space!
+