Update app.py

app.py

@@ -23,15 +23,15 @@ prompt_template = "Responda as questões.\nHuman: {prompt}\nAssistant:\n"
 def get_response(text):
     prompt = prompt_template.format(prompt=text)
     response = lcpp_llm(
-        prompt=prompt,
-        max_tokens=256,
-        temperature=0.5,
-        top_p=0.95,
-        top_k=50,
-        stop=[''], # Dynamic stopping when such token is detected.
-        echo=True # return the prompt
-    )
-    return response
+    prompt=prompt,
+    max_tokens=256,
+    temperature=0.5,
+    top_p=0.95,
+    top_k=50,
+    stop = ['<|end_of_text|>'], # Dynamic stopping when such token is detected.
+    echo=True # return the prompt
+)
+    return response['choices'][0]['text'].split('Assistant:\n')[1]
 
 interface = gr.Interface(
     fn=get_response,