Update app.py

app.py

@@ -55,8 +55,13 @@ def predict(img):
     gray_img = pil_image.convert('L')#grayscale
     pic = np.array(gray_img) #convert to array
     inp_img=reduce_image_count(pic)#Reduce image to required input size
-    
-    otpt=F.softmax(mdlV2(Tensor(inp_img.view(-1,64))))[0].data
+    z=Tensor(inp_img)
+    y=z.view(-1,64)
+    x=mdlV2(y)
+    w=F.softmax(x)
+    v=w[0]
+    u=v.data
+    otpt=u
     #pred,pred_idx,probs = learn.predict(img)
     
     return dict([[labels[i], float(otpt[i])] for i in range(len(labels))]),inp_img