Update app.py

app.py

@@ -3,13 +3,13 @@ from transformers import pipeline
 from transformers import TextDataset, DataCollatorForLanguageModeling
 from transformers import Trainer, TrainingArguments,AutoModelWithLMHead
 
-chef = pipeline('text-generation', model="./en_gpt2-medium_rachel_replics/en_gpt2-medium_rachel_replics", tokenizer=model_type)
+chef = pipeline('text-generation', model="./en_gpt2-medium_rachel_replics/en_gpt2-medium_rachel_replics", tokenizer="gpt2-medium")
 
 
 # gradio part 
 def echo(message, history, model):
     
-    chef = pipeline('text-generation', model="./models/en_gpt2-medium_rachel_replics", tokenizer=model_type)
+    #chef = pipeline('text-generation', model="./models/en_gpt2-medium_rachel_replics", tokenizer=model_type)
     
     if model=="gpt2-medium":
         answer = chef(f"<s>NOTFRIEND: {message}\nRACHEL:")[0]['generated_text']