Update app.py

app.py

@@ -40,9 +40,11 @@ def response(image):
         elif cl == 1:
             name_weap += name[cl] + '\n'
         elif cl == 2:
-            name_weap += model2(image[0]["label"]) + '\n'
-        elif cl == 3:
-            name_weap += model2(image[0]["label"]) + '\n'
+            out = model2(image)
+            name_weap += out[0]["label"] + '\n'
+        elif cl == 3
+            out = model2(image)
+            name_weap += out[0]["label"] + '\n'
 
         
     # im_rgb = Image.fromarray(im_rgb)