Update app.py

app.py

@@ -5,6 +5,7 @@ import spaces
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 import os
 from threading import Thread
+from pydantic import BaseModel
 
 import pymupdf
 import docx