Update app.py

app.py

@@ -56,9 +56,9 @@ def predict(img):
     pic = np.array(gray_img) #convert to array
     inp_img=reduce_image_count(pic)#Reduce image to required input size
     
-    otpt=F.softmax(mdlV2(inp_img.view(-1,64)))
+    otpt=F.softmax(mdlV2(Tensor(inp_img.view(-1,64))))[0].data
     #pred,pred_idx,probs = learn.predict(img)
     
-    return dict([[labels[i], float(otpt[0].data[i])] for i in range(len(labels))]),inp_img
+    return dict([[labels[i], float(otpt[i])] for i in range(len(labels))]),inp_img
 
 gr.Interface(fn=predict, inputs=gr.inputs.Image(type='filepath'),   outputs=[gr.outputs.Label(num_top_classes=10), gr.outputs.Image()]).launch(share=True)