KikoDM commited on
Commit
3c55658
·
1 Parent(s): 1394c4f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -13
app.py CHANGED
@@ -53,21 +53,21 @@ def predict_image(pic):
53
  preds = loaded_model.predict(x)
54
  preds = list(preds[0])
55
 
56
- #list_desc_order = heapq.nlargest(2, range(len(preds)), key=preds.__getitem__)
57
- #result1 = f'c{list_desc_order[0]}'
58
- #result2 = '-'
59
- #result2_ = 0
60
- #if preds[list_desc_order[1]] > 0.3:
61
- # result2 = f'c{list_desc_order[1]}'
62
- # result2_ = round(preds[list_desc_order[1]], 2)
63
- #txt = f"category {directory} result 1 {result1} {round(preds[list_desc_order[0]],2)} | result2 {result2} {result2_}"
64
- #txt = f"categoria {directory}"
65
 
66
- #score = round(preds[list_desc_order[0]], 2)*100
67
- #score = int(score)
68
- #txt2 = f"resultado: {class_dict.get(result1)} probabilidad {score}%"
69
  txt3="pepe"
70
- return txt3
71
 
72
 
73
  iface = gr.Interface(
 
53
  preds = loaded_model.predict(x)
54
  preds = list(preds[0])
55
 
56
+ list_desc_order = heapq.nlargest(2, range(len(preds)), key=preds.__getitem__)
57
+ result1 = f'c{list_desc_order[0]}'
58
+ result2 = '-'
59
+ result2_ = 0
60
+ if preds[list_desc_order[1]] > 0.3:
61
+ result2 = f'c{list_desc_order[1]}'
62
+ result2_ = round(preds[list_desc_order[1]], 2)
63
+ txt = f"category {directory} result 1 {result1} {round(preds[list_desc_order[0]],2)} | result2 {result2} {result2_}"
64
+ txt = f"categoria {directory}"
65
 
66
+ score = round(preds[list_desc_order[0]], 2)*100
67
+ score = int(score)
68
+ txt2 = f"resultado: {class_dict.get(result1)} probabilidad {score}%"
69
  txt3="pepe"
70
+ return txt2
71
 
72
 
73
  iface = gr.Interface(