Fix

Dockerfile

@@ -6,8 +6,6 @@ COPY ./requirements.txt /code/requirements.txt
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
-RUN git clone https://huggingface.co/GuysTrans/bart-base-finetuned-xsum
-
 COPY . .
 
 CMD ["flask", "run", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -5,13 +5,9 @@ from huggingface_hub.hf_api import HfFolder
 from transformers import AutoModelForSeq2SeqLM
 from transformers import AutoTokenizer
 
-HfFolder.save_token('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')
+tokenizer = AutoTokenizer.from_pretrained("facebook/bart-base")
 
-model_checkpoint = "facebook/bart-base"
-model_name = model_checkpoint.split("/")[-1]
-tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
-
-model = AutoModelForSeq2SeqLM.from_pretrained(f"{model_name}-finetuned-xsum")
+model = AutoModelForSeq2SeqLM.from_pretrained("GuysTrans/bart-base-finetuned-xsum")
 
 
 def generate_summary(question, model):