unit tested and corrected the runtime errors

app.py

@@ -1,5 +1,4 @@
 from pathlib import Path
-import torch
 import gradio as gr
 from huggingface_hub import from_pretrained_fastai
 
@@ -13,9 +12,7 @@ def predict(news_headline):
 
     probabilities = learner.predict(news_headline)
 
-    # values, indices = torch.topk(probabilities[0]['probs'], 5)
-
-    return {LABELS[i]: probabilities[0]['probs'][i] for i, v in range(len(LABELS))}
+    return {LABELS[i]: probabilities[0]['probs'][i] for i in range(len(LABELS))}
 
 interface = gr.Interface(
     predict,