Update app.py

app.py

@@ -5,10 +5,11 @@ import gradio as gr
 from llama_index.readers.file import PDFReader
 
 ### added to remove openapi
-import torch
+from transformers import AutoModelForCausalLM
+
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model_name = "baffo32/decapoda-research-llama-7B-hf"
-model = torch.hub.load("HuggingFace", "transformers", model_name).to(device)
+model = AutoModelForCausalLM.from_pretrained(model_name).to(device)
 
 ### added to remove openapi