Update main.py

main.py

@@ -1,3 +1,4 @@
+import torch
 from transformers import pipeline 
 generate_text = pipeline(model="databricks/dolly-v2-2-8b",
                          torch_dtype=torch.bfloat16,