Update app.py

app.py

@@ -1167,11 +1167,12 @@ def display_historical_data(storage_location: str, url: str):
 # Function to load the "google/flan-t5-xl" model
 def load_model():
     """
-    Loads the FlanT5XL model and tokenizer once and returns the pipeline.
+    Loads the openLlama model and tokenizer once and returns the pipeline.
     """
-        model_name = "openlm-research/open_llama_3b_v2"
-tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False, legacy=False)
-model = AutoModelForCausalLM.from_pretrained(model_name)  # Use AutoModelForCausalLM
+    try:
+        model_name = "openlm-research/open_llama_3b_v2"  # Correct indentation here
+        tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False, legacy=False)
+        model = AutoModelForCausalLM.from_pretrained(model_name) 
 
 # Determine the maximum supported length for the model
 max_supported_length = 2048  # You might need to adjust this