change models endpoints

app.py

@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, TFAutoModelForCausalLM, TFAutoModelForSe
 
 # MODEL TO CALL
 
-generator_name = "Alirani/distilgpt2-finetuned-synopsis-genres_final"
+generator_name = "Alirani/synopsis_generator"
 tokenizer_gen = AutoTokenizer.from_pretrained(generator_name)
 model_gen = TFAutoModelForCausalLM.from_pretrained(generator_name)
 
@@ -15,7 +15,7 @@ def generate_synopsis(model, tokenizer, title):
     processed_synopsis = "".join(synopsis.split('|')[2].rpartition('.')[:2]).strip()
     return processed_synopsis
 
-classifier_name = "Alirani/overview_classifier_final"
+classifier_name = "Alirani/synopsis_classifier"
 tokenizer_clf = AutoTokenizer.from_pretrained(classifier_name)
 model_clf = TFAutoModelForSequenceClassification.from_pretrained(classifier_name)