Update app.py

Fix bugs in load_model

app.py

@@ -4,10 +4,12 @@ from transformers import T5Tokenizer, T5ForConditionalGeneration
 tokenizer = T5Tokenizer.from_pretrained("jokes-tokenizer")
 
 @st.cache
-def load_model(model_name):
+def load_model():
     model = T5ForConditionalGeneration.from_pretrained("jokes-model")
     return model
 
+model = load_model()
+
 def infer(input_ids):
     output_sequences = model.generate(input_ids=input_ids)
     return tokenizer.decode(outputs[0], skip_special_tokens=True)