Update app.py

app.py

@@ -92,12 +92,14 @@ def generate(
     else :
         base_prompt = NO_PROMPT
 
-    chat_prompt = prompt + "\n\nAnswer:"
-    prompt = base_prompt + chat_prompt
 
     if version == "StarCoder-gradio" :
+        chat_prompt = prompt + "\n\nAnswer:"
+        prompt = base_prompt + chat_prompt
         stream = client_g.generate_stream(prompt, **generate_kwargs)
     elif version == "StarChat-alpha" :
+        chat_prompt = prompt + "\n\nAssistant:"
+        prompt = base_prompt + chat_prompt
         stream = client_s.generate_stream(prompt, **generate_kwargs)
     else :
         pass