Update app.py

app.py

@@ -1,19 +1,19 @@
 import streamlit as st
-from transformers import pipeline, AutoModel
-import torch
+from transformers import pipeline
+
+MODEL_PATH = "zephyr_7b_beta_model"
 
 def main():
     st.title("Chatbot with Hugging Face Model")
 
     # Check if the model is already saved locally
-    model_path = "./zephyr-7b-beta"
     try:
-        pipe = pipeline("text-generation", model=model_path, torch_dtype=torch.bfloat16, device_map="auto")
+        pipe = pipeline("text-generation", model=MODEL_PATH, torch_dtype=torch.bfloat16, device_map="auto")
     except:
         # If not saved, load the model and save it
         st.warning("Model not found locally. Downloading and saving the model. Please wait...")
         pipe = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta", torch_dtype=torch.bfloat16, device_map="auto")
-        pipe.save_pretrained(model_path)
+        pipe.save_pretrained(MODEL_PATH)
 
     # Define chat messages
     messages = [