Update app.py

app.py

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