Add torch

app.py

@@ -7,6 +7,7 @@ import gradio as gr
 # iface.launch()
 
 import spaces
+import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, Trainer, TrainingArguments, pipeline, set_seed
 
 device = "cuda:0" if torch.cuda.is_available() else "cpu"