Update app.py

app.py

@@ -9,7 +9,7 @@ from uuid import uuid4
 import gradio as gr
 import spaces
 import torch
-from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
+from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer, AutoModel
 from pathlib import Path
 from pinecone.grpc import PineconeGRPC as Pinecone
 import torch