logging values

app.py

@@ -7,6 +7,7 @@ sign_api = api()
 def sign(input_img):
     input_img = Image.fromarray(input_img)
     prediction = sign_api.predict(input_img)
+    print('prediction',prediction)
     return prediction
 
 demo = gr.Interface(sign,inputs=gr.Image(shape=(200, 200)), outputs=gr.Label())

model.py

@@ -74,6 +74,7 @@ class api():
                 print('ArgMax',ArgMax_val)
                 index = ArgMax_val.tolist()[0]
                 class_val = self.classes[index]
+                print('class_val',class_val)
 
                 # os.remove(os.path.join(self.upload_path,filename))