Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, pipeline, AutoConfig
 from huggingface_hub import cached_download, hf_hub_url, list_models
+from transformers.modeling_utils import PreTrainedModel
 import requests
 import json
 import os