rajyalakshmijampani commited on
Commit
eb9d6db
·
1 Parent(s): e628897

logging updated

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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
  # -------------------