Spaces:

StKirill
/

GenerativeChatBot

Sleeping

StKirill commited on Mar 2, 2024

Commit

5ceb303

verified ·

1 Parent(s): 7353ca0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,17 +12,17 @@ def echo(message, history, model):
     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']
         answer = tmp[tmp.find(f"RACHEL: ") + len("RACHEL") + 2 : tmp.find('</s>')]
         return answer
     elif model=="gpt2-medium":
-        answer = chef(f"<s>NOTFRIEND: {message}}\nRACHEL:")[0]['generated_text']
         answer = tmp[tmp.find(f"RACHEL: ") + len("RACHEL") + 2 : tmp.find('</s>')]
         return answer
     elif model=="gpt2-medium":
-        answer = chef(f"<s>NOTFRIEND: {message}}\nRACHEL:")[0]['generated_text']
         answer = tmp[tmp.find(f"RACHEL: ") + len("RACHEL") + 2 : tmp.find('</s>')]
         return answer

     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']
         answer = tmp[tmp.find(f"RACHEL: ") + len("RACHEL") + 2 : tmp.find('</s>')]
         return answer
     elif model=="gpt2-medium":
+        answer = chef(f"<s>NOTFRIEND: {message}\nRACHEL:")[0]['generated_text']
         answer = tmp[tmp.find(f"RACHEL: ") + len("RACHEL") + 2 : tmp.find('</s>')]
         return answer
     elif model=="gpt2-medium":
+        answer = chef(f"<s>NOTFRIEND: {message}\nRACHEL:")[0]['generated_text']
         answer = tmp[tmp.find(f"RACHEL: ") + len("RACHEL") + 2 : tmp.find('</s>')]
         return answer