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):
|