Spaces:

StKirill
/

GenerativeChatBot

Running

StKirill commited on Mar 2

Commit

70af8cf

•

1 Parent(s): 94daf08

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,17 +14,17 @@ def echo(message, history, model):
     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

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