Update app.py

app.py

@@ -46,12 +46,12 @@ def generate(model, tokenizer, test_samples, prefix="", max_length=256):
 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 = ""
@@ -72,6 +72,6 @@ The current sources supported for classification are: The Times of India, The In
 placeholder = "Enter the content of the article here."
 
 iface = gradio.Interface(fn=main, inputs=[gradio.inputs.Textbox(lines=10, placeholder=placeholder, label='Article Content:'), 
-                                          gradio.inputs.Checkbox(default=True, label='Classify the Source:')], outputs="textbox", title=title, 
-                         description=description)
+                                  gradio.inputs.Checkbox(default=True, label='Classify the Source:')], outputs="textbox", title=title, 
+                 description=description)
 iface.launch()