Update app.py

app.py

@@ -52,8 +52,8 @@ def aiornot0(image):
     for idx,result in enumerate(px):
         results[labels[idx]] = px[idx][0]
     #results[labels['label']] = result['score']
-    tmp_res={f'{uid}-0':results}
-    fin_sum.append(tmp_res)
+    #tmp_res={f'{uid}-0':results}
+    fin_sum.append(results)
     return gr.HTML.update(html_out),results
 def aiornot1(image):    
     labels = ["Real", "AI"]
@@ -78,8 +78,8 @@ def aiornot1(image):
     for idx,result in enumerate(px):
         results[labels[idx]] = px[idx][0]
     #results[labels['label']] = result['score']
-    tmp_res={f'{uid}-1':results}
-    fin_sum.append(tmp_res)        
+    #tmp_res={f'{uid}-1':results}
+    fin_sum.append(results)        
     return gr.HTML.update(html_out),results    
 def aiornot2(image):    
     labels = ["AI", "Real"]
@@ -104,8 +104,8 @@ def aiornot2(image):
     results = {}
     for idx,result in enumerate(px):
         results[labels[idx]] = px[idx][0]
-    tmp_res={f'{uid}-2':results}
-    fin_sum.append(tmp_res)
+    #tmp_res={f'{uid}-2':results}
+    fin_sum.append(results)
     return gr.HTML.update(html_out),results
 def aiornot3(image):    
     labels = ["Real", "AI"]
@@ -194,7 +194,7 @@ def load_url(url):
 
 def tot_prob():
     try:
-        fin_out = fin_sum[0][0]+fin_sum[1][0]+fin_sum[2][0]
+        fin_out = fin_sum[0]["Real"]+fin_sum[1]["Real"]+fin_sum[2]["Real"]
         print (fin_out)
     except Exception as e:
         print (f'ERROR :: {e}')