Update app.py

app.py

@@ -51,7 +51,6 @@ Content: "{content}"
     log.info(f"🧠 Inferência iniciada para: {title}")
     start_time = time.time()
 
-    # Tokenização + attention mask
     inputs = tokenizer(
         prompt,
         return_tensors="pt",
@@ -68,8 +67,8 @@ Content: "{content}"
             attention_mask=attention_mask,
             max_new_tokens=100,
             temperature=1.0,
-            do_sample=False,  # Greedy decoding
-            top_k=50,         # Razoável para limitar
+            do_sample=False,
+            top_k=50,
             no_repeat_ngram_size=2,
             num_beams=1,
             eos_token_id=tokenizer.eos_token_id,
@@ -90,7 +89,7 @@ Content: "{content}"
         return json_result
     else:
         log.warning("⚠️ Falha ao extrair JSON.")
-        return "⚠️ Failed to extract JSON. Output:\n" + generated
+        raise HTTPException(status_code=404, detail="Unable to extract JSON from model output.")
 
 # -------- API --------
 @app.get("/filter")
@@ -102,9 +101,8 @@ def get_filter(
         json_output = infer_filter(title, content)
         import json
         return json.loads(json_output)
-    except json.JSONDecodeError:
-        log.error("❌ Erro ao fazer parse do JSON.")
-        return {"raw_output": json_output}
+    except HTTPException as he:
+        raise he  # já tratado
     except Exception as e:
         log.exception("❌ Erro inesperado:")
-        raise HTTPException(status_code=422, detail=str(e))
+        raise HTTPException(status_code=404, detail="Invalid or malformed JSON output from model.")