Update app.py

app.py

@@ -1,27 +1,8 @@
-import os
-import gdown
 from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
 import gradio as gr
 
-# Define Google Drive file IDs and corresponding file names
-files = {
-    'vocab.json': '1b0lO8qDUE6N36yXZzokpCTCG-13GO8Nx',
-    'merges.txt': '1R2W6k7iHlkzoxLmd1pMQZ9Y7phHjaLFk',
-    'config.json': '1U5PzBzLyWhs5Z1smW5w7P-PsJZJ_JB1c',
-    'generation_config.json': '1GqkR8f_TMCN8Qls4YmK5E7YtP8V9IW27',
-    'special_tokens_map.json': '1R4MkInjvH_u6IrW_RR1vc_R5nUQV-J5S',
-    'tokenizer_config.json': '1l4OGiYwJ4GR9RMIvMX8f8xSkbMC_76v7',
-    'training_args.bin': '1YPDBwO6Q1cszH_K1bVwXkZn7BtI5G2Ex'
-}
-
-# Download files from Google Drive
-for file_name, file_id in files.items():
-    url = f'https://drive.google.com/uc?id={file_id}'
-    output = file_name
-    gdown.download(url, output, quiet=False)
-
 # Load the model and tokenizer
-model_path = '.'  # Path to the current directory where files are downloaded
+model_path = '.'  # Path to the current directory where files are located
 
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_path)