Update src/prediction.py

src/prediction.py

@@ -2,7 +2,7 @@ from transformers import FlaxAutoModelForSeq2SeqLM
 from transformers import AutoTokenizer
 import textwrap
 
-MODEL_NAME_OR_PATH = "JustAPR/t5-recipe-generation"
+MODEL_NAME_OR_PATH = "JustAPR/resGen"
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME_OR_PATH, use_fast=True)
 model = FlaxAutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME_OR_PATH)