Update app.py

app.py

@@ -29,7 +29,10 @@ def query(system_prompt, user_prompt, temperature=0.9, max_new_tokens=256, top_p
         do_sample=True,
         seed=seed,
     )
-    prompt = f"System: {system_prompt}\nUser: {user_prompt}\nFalcon:"
+    print(system_prompt)
+    print(user_prompt)
+    print('-' * 20)
+    prompt = f"System: {system_prompt}\nUser: {user_prompt}\Medic:"
     stream = client.text_generation(prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""
     for response in stream:
@@ -41,6 +44,8 @@ def query(system_prompt, user_prompt, temperature=0.9, max_new_tokens=256, top_p
                 output = output.rstrip()
                 #yield output
         #yield output
+    print(output)
+    print('-' * 20)
     return output
 
 iface = gr.Interface(