Update app.py

app.py

@@ -28,15 +28,8 @@ def CustomChatGPT(Ask):
                   repeat_penalty=1.2, top_k=150,
                   echo=True)
     
-    reply=response["choices"][0]["text"]
-    # Split the response by lines
-    #response_lines = reply.split("\n")
-
-    # Find the index of the ASSISTANT line
-    #assistant_index = response_lines.index("ASSISTANT:")
-
-    # Extract the assistant response by joining the lines after the ASSISTANT line
-    #assistant_response = "\n".join(response_lines[assistant_index + 1:])
+    #reply=response["choices"][0]["text"]
+    reply = response["choices"][0]["text"].split("ASSISTANT:")[1].strip()
     return reply
 
 gui=gr.Interface(fn=CustomChatGPT, inputs="text", outputs="text", examples=["Who is the greatest basketball player in NBA history","What is the winning record in a season?"] , title="Ask the AI coach", description="Ask the AI coach all you want about NBA Teams and Players:")