Update app.py

app.py

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