Update app.py

app.py

@@ -3,10 +3,8 @@ from transformers import pipeline
 from transformers import TextDataset, DataCollatorForLanguageModeling
 from transformers import Trainer, TrainingArguments,AutoModelWithLMHead
 
-chef = pipeline('text-generation', model="./models/en_gpt2-medium_rachel_replics", tokenizer=model_type)
-tmp = chef(f"<s>NOTFRIEND: Do you like Ross?\nRACHEL:")[0]['generated_text']
-tmp = tmp[tmp.find(f"RACHEL: ") + len("RACHEL") + 2 : tmp.find('</s>')]
-tmp
+chef = pipeline('text-generation', model="./en_gpt2-medium_rachel_replics/en_gpt2-medium_rachel_replics", tokenizer=model_type)
+
 
 # gradio part 
 def echo(message, history, model):