Update app.py

app.py

@@ -64,11 +64,11 @@ def handle_query(query):
     try:
         data=json.loads(response.content.decode("utf-8"))
         result=data[0]['generated_text']
-        last_comma_index = result.rfind('assistant<|end_header_id|>')
+        last_comma_index = result.rfind('|eot_id|>assistant')
 
         # If a comma is found, slice the string from the last comma onwards
         if last_comma_index != -1:
-            result = result[last_comma_index + 26:].strip()
+            result = result[last_comma_index + 18:].strip()
             
         return result
     except: