Update app.py
Browse files
app.py
CHANGED
|
@@ -44,7 +44,7 @@ print(f"Successfully loaded the model {model_name} into memory")
|
|
| 44 |
|
| 45 |
|
| 46 |
start_message = """A chat between a curious human and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions."""
|
| 47 |
-
prompt = f"{
|
| 48 |
|
| 49 |
|
| 50 |
class StopOnTokens(StoppingCriteria):
|
|
|
|
| 44 |
|
| 45 |
|
| 46 |
start_message = """A chat between a curious human and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions."""
|
| 47 |
+
prompt = f"{start_message} ### Human: {user_query} ### Assistant:"
|
| 48 |
|
| 49 |
|
| 50 |
class StopOnTokens(StoppingCriteria):
|