Update app.py

app.py

@@ -3,16 +3,16 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 import pandas as pd
 import re
 
-# Load the LLaMA model and tokenizer
+# Load the Falcon-7B-Instruct model and tokenizer
 @st.cache_resource
-def load_llama_model():
-    """Load the LLaMA model and tokenizer."""
-    model_name = "meta-llama/Llama-2-7b-chat-hf"  # Replace with your preferred LLaMA model
+def load_falcon_model():
+    """Load the Falcon-7B-Instruct model and tokenizer."""
+    model_name = "tiiuae/falcon-7b-instruct"  # Open model for instruction-based tasks
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForCausalLM.from_pretrained(model_name)
     return pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-qa_pipeline = load_llama_model()
+qa_pipeline = load_falcon_model()
 
 # Load SOP Dataset
 @st.cache_data
@@ -39,7 +39,7 @@ def find_best_context(question, dataset):
     return best_context
 
 # Streamlit UI
-st.title("SOP Question Answering AI with LLaMA")
+st.title("SOP Question Answering AI with Falcon")
 st.markdown("Ask any question about Standard Operating Procedures:")
 
 # User input