Update app.py

app.py

@@ -2,9 +2,10 @@ import streamlit as st
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-# Load the Mistral-7B-Instruct model
-tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
-model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1", torch_dtype=torch.float16, device_map="auto")
+# Load the Zephyr-7B-Alpha model (fully open and optimized for instruction-following)
+MODEL_NAME = "HuggingFaceH4/zephyr-7b-alpha"
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.float16, device_map="auto")
 
 # Initialize conversation history if not present
 if "conversation" not in st.session_state: