Update app.py

app.py

@@ -5,7 +5,7 @@ import gradio as gr
 from llama_index.readers.file import PDFReader
 
 ### added to remove openapi
-
+import torch
 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)