Update app.py

app.py

@@ -24,8 +24,9 @@ def text_inference(text, language):
     output_ids = model.generate(input_ids, max_new_tokens=256)
     output_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
 
-    # Parse comma-separated string into list
-    words = [word.strip() for word in output_text.split(",") if word.strip()]
+     # Parse response: take last line, split by commas
+    last_line = output_text.strip().split("\n")[-1]
+    words = [w.strip() for w in last_line.split(",") if w.strip()]
     return words
 
 def ocr_inference(img, lang):