updated app.py

app.py

@@ -37,8 +37,9 @@ def predict(image, model, device):
     with torch.no_grad():
         outputs = model(image_t)
         probs = torch.softmax(outputs, dim=1)
-        conf, predicted = torch.max(probs, 1)
-    return class_names[predicted.item()], conf.item()
+        conf, predicted = torch.max(probs, dim=1)
+    return class_names[predicted.item()], float(conf.item()) # type: ignore
+
 
 def main():
     st.title("Human Action Recognition App")