Fixes for certain imports

app.py

@@ -1,4 +1,6 @@
 from pathlib import Path
+import torch
+import gradio as gr
 from huggingface_hub import from_pretrained_fastai
 
 
@@ -9,7 +11,7 @@ LABELS = Path('class_names.txt').read_text().splitlines()
 def predict(im):
     learner = from_pretrained_fastai("rajeshradhakrishnan/ml-news-classify-fastai")
 
-    probabilities = learner.predict(sentences)
+    probabilities = learner.predict(im)
 
     values, indices = torch.topk(probabilities, 5)
 
@@ -20,8 +22,8 @@ interface = gr.Interface(
     inputs="newsheadlines", 
     outputs='label', 
     theme="huggingface", 
-    title="Sketch Recognition", 
-    description="Malayalam News Classifier? Input a few malayalam news headlines and verify whether the model categorized it apporpirately!", 
+    title="Malayalam News Classifier", 
+    description="Try to classify news in മലയാളം? Input a few malayalam news headlines and verify whether the model categorized it apporpirately!", 
     article = "<p style='text-align: center'>Malayalam News Classifier | Demo Model</p>",
     live=True)
 interface.launch(debug=True)