Update app.py

app.py

@@ -7,7 +7,7 @@ import torch
 from utils import *
 from presets import *
 import time
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
 
 ######################################################################