Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
-from transformers import LlamaTokenizer, LlamaForCausalLM
+from transformers import AutoModelForCausalLM, AutoTokenizer
 import tempfile
 import numpy as np
 
 # Initialize LLaMA Model for Question Answering
-llama_tokenizer = LlamaTokenizer.from_pretrained('huggingface/llama-7b')
-llama_model = LlamaForCausalLM.from_pretrained('huggingface/llama-7b')
+llama_model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
+llama_tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
 
 # Updated transcribe_and_predict_video function from your code
 def transcribe_and_predict_video(video):