Commit
·
eb9d6db
1
Parent(s):
e628897
logging updated
Browse files
app.py
CHANGED
|
@@ -122,13 +122,12 @@ def classify_text(claim):
|
|
| 122 |
evidences = get_evidence_sentences(claim)
|
| 123 |
evidence_text = " ".join(evidences)
|
| 124 |
|
| 125 |
-
print(evidence_text)
|
| 126 |
-
|
| 127 |
# Step 1: FEVER classification
|
| 128 |
text = f"claim: {claim} evidence: {evidence_text}"
|
| 129 |
result = classifier(text, truncation=True, max_length=512, return_all_scores=True)[0]
|
| 130 |
top_label = sorted(result, key=lambda x: x["score"], reverse=True)[0]["label"]
|
| 131 |
label_str = "REAL" if top_label == "LABEL_0" else "FAKE"
|
|
|
|
| 132 |
|
| 133 |
# Step 2: Mistral explanation generation
|
| 134 |
prompt = f"""
|
|
@@ -162,7 +161,7 @@ def classify_text(claim):
|
|
| 162 |
completion = inf_client.chat_completion( model=explain_model, messages=messages, max_tokens=256, temperature=0.3)
|
| 163 |
explanation = completion.choices[0].message.content.strip()
|
| 164 |
|
| 165 |
-
return f"Explanation:\n{explanation}"
|
| 166 |
|
| 167 |
|
| 168 |
# -------------------
|
|
|
|
| 122 |
evidences = get_evidence_sentences(claim)
|
| 123 |
evidence_text = " ".join(evidences)
|
| 124 |
|
|
|
|
|
|
|
| 125 |
# Step 1: FEVER classification
|
| 126 |
text = f"claim: {claim} evidence: {evidence_text}"
|
| 127 |
result = classifier(text, truncation=True, max_length=512, return_all_scores=True)[0]
|
| 128 |
top_label = sorted(result, key=lambda x: x["score"], reverse=True)[0]["label"]
|
| 129 |
label_str = "REAL" if top_label == "LABEL_0" else "FAKE"
|
| 130 |
+
print(f"[INFO] Model Classified {claim} as {label_str}")
|
| 131 |
|
| 132 |
# Step 2: Mistral explanation generation
|
| 133 |
prompt = f"""
|
|
|
|
| 161 |
completion = inf_client.chat_completion( model=explain_model, messages=messages, max_tokens=256, temperature=0.3)
|
| 162 |
explanation = completion.choices[0].message.content.strip()
|
| 163 |
|
| 164 |
+
return f"Explanation:\n\n{explanation}"
|
| 165 |
|
| 166 |
|
| 167 |
# -------------------
|