Update app.py

app.py

@@ -38,10 +38,10 @@ assistant
         generated_text = output[0].get('generated_text', '')
         print("Generated Text:", generated_text)  # Print the generated text
 
-        # Extract the first non-empty line as the category
-        lines = [line.strip().lower() for line in generated_text.split('\n') if line.strip()]
-        if lines:
-            return lines[0]
+        # Extract the response after the last 'assistant' keyword
+        parts = generated_text.split('assistant\n')
+        if len(parts) > 1:
+            return parts[-1].strip().lower()  # Extract the text after the last 'assistant'
     return "unknown"
 
 # Entrée personnalisée pour le test