Update app.py

app.py

@@ -1,6 +1,5 @@
-import transformers
 import re
-from transformers import AutoConfig, AutoTokenizer, AutoModel, AutoModelForCausalLM
+from transformers import AutoModelForCausalLM, AutoTokenizer, AutoConfig, AutoModel, StoppingCriteria, StoppingCriteriaList, TextIteratorStreamer
 from vllm import LLM, SamplingParams
 import torch
 import gradio as gr