Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -92,12 +92,14 @@ def generate(
|
|
92 |
else :
|
93 |
base_prompt = NO_PROMPT
|
94 |
|
95 |
-
chat_prompt = prompt + "\n\nAnswer:"
|
96 |
-
prompt = base_prompt + chat_prompt
|
97 |
|
98 |
if version == "StarCoder-gradio" :
|
|
|
|
|
99 |
stream = client_g.generate_stream(prompt, **generate_kwargs)
|
100 |
elif version == "StarChat-alpha" :
|
|
|
|
|
101 |
stream = client_s.generate_stream(prompt, **generate_kwargs)
|
102 |
else :
|
103 |
pass
|
|
|
92 |
else :
|
93 |
base_prompt = NO_PROMPT
|
94 |
|
|
|
|
|
95 |
|
96 |
if version == "StarCoder-gradio" :
|
97 |
+
chat_prompt = prompt + "\n\nAnswer:"
|
98 |
+
prompt = base_prompt + chat_prompt
|
99 |
stream = client_g.generate_stream(prompt, **generate_kwargs)
|
100 |
elif version == "StarChat-alpha" :
|
101 |
+
chat_prompt = prompt + "\n\nAssistant:"
|
102 |
+
prompt = base_prompt + chat_prompt
|
103 |
stream = client_s.generate_stream(prompt, **generate_kwargs)
|
104 |
else :
|
105 |
pass
|