Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import time
 from transformers import AutoModelForCausalLM  # Please ensure this import is correct
+from download_model import download_model
 
 PROMPT_TEMPLATE = (
     "<s>" "[INST]"
@@ -18,7 +19,7 @@ PROMPT_TEMPLATE = (
 )
 
 def load_llm():
-    llm = AutoModelForCausalLM.from_pretrained("atwine/Llama-2-7b-chat-q8-gguf",
+    llm = AutoModelForCausalLM.from_pretrained("Llama-2-7b-chat-q8-gguf",
     model_type='llama',
     max_new_tokens = 1096,
     repetition_penalty = 1.13,