Update app.py

app.py

@@ -43,15 +43,15 @@ def generate(model, tokenizer, test_samples, prefix="", max_length=256):
         output_str = tokenizer.batch_decode(outputs, skip_special_tokens=True)
         return output_str[0]
  
- def classify(model, tokenizer, content, title):
+def classify(model, tokenizer, content, title):
     model.eval()
     with torch.no_grad():
-        model_inputs = tokenizer(title, content, padding=True, truncation=True, return_tensors="pt").to(device)
-        outputs = model(**model_inputs)
-        logits = outputs.logits
-        selected = logits.argmax(dim=-1).cpu().tolist()
-        answers = [rev_mapper[sel] for sel in selected]
-        return answers[0]
+    model_inputs = tokenizer(title, content, padding=True, truncation=True, return_tensors="pt").to(device)
+    outputs = model(**model_inputs)
+    logits = outputs.logits
+    selected = logits.argmax(dim=-1).cpu().tolist()
+    answers = [rev_mapper[sel] for sel in selected]
+    return answers[0]
         
 def main(content, classify_source=False):
     output = ""