Import torch and add as requirement

app.py

@@ -1,4 +1,5 @@
 from fastai.vision.all import *
+import torch
 import gradio as gr
 
 learn = load_learner("model_tvdesktop.pkl")
@@ -11,7 +12,7 @@ def classify_image(img):
     print(probs[0]*100)
     for out in output:
         val = output[out]
-        if val < 60:
+        if val.item()*100 < 60:
             return {"Not Sure/Others": 0}
     return output
 

requirements.txt

@@ -1 +1,2 @@
 fastai
+torch