Mehrdad Esmaeili commited on
Commit
a3bdb8c
·
1 Parent(s): 18c2744

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,8 +30,8 @@ class StopOnTokens(StoppingCriteria):
30
  return False
31
 
32
  def predict(message, history):
33
- #prompt,userInput = message.split('!!')
34
- #message=generate_prompt({"instruction": prompt, "input": userInput})
35
  history_transformer_format = history + [[message, ""]]
36
  stop = StopOnTokens()
37
 
 
30
  return False
31
 
32
  def predict(message, history):
33
+ prompt,userInput = message.split('!!')
34
+ message=generate_prompt({"instruction": prompt, "input": userInput})
35
  history_transformer_format = history + [[message, ""]]
36
  stop = StopOnTokens()
37