Update app.py

app.py

@@ -33,6 +33,7 @@ topic_model = pipeline("zero-shot-classification", model="valhalla/distilbart-mn
 #model = pipeline("text-generation", model="Organika/StarCoder-7B-WoW-JSON_1", device=0)
 
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
+import torch
 
 bnb_config = BitsAndBytesConfig(
 load_in_4bit=True,