Update app.py

app.py

@@ -5,7 +5,7 @@ from datetime import datetime
 from theme import TufteInspired
 from transformers import AutoTokenizer
 from transformers import pipeline
-
+import torch
 
 from huggingface_hub import login
 import os