Spaces:
Runtime error
Runtime error
Mehrdad Esmaeili
commited on
Commit
·
a3bdb8c
1
Parent(s):
18c2744
Update app.py
Browse files
app.py
CHANGED
|
@@ -30,8 +30,8 @@ class StopOnTokens(StoppingCriteria):
|
|
| 30 |
return False
|
| 31 |
|
| 32 |
def predict(message, history):
|
| 33 |
-
|
| 34 |
-
|
| 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 |
|