Update app.py

app.py

@@ -1,3 +1,4 @@
+import gradio as gr
 import os
 from huggingface_hub import HfApi, login
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
@@ -7,7 +8,7 @@ def process(model_id, dataset):
     tokenizer = AutoTokenizer.from_pretrained(model_id)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_id)
     # Local Path of Model
-    model_path = 't5-fine-tune-save-example'
+    model_path = model_id
     model.save_pretrained(model_path)
     login(token=os.environ["HF_TOKEN"])
     api = HfApi()