citradiani commited on
Commit
70e3507
·
verified ·
1 Parent(s): 4d69d16

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -8
app.py CHANGED
@@ -77,14 +77,23 @@ def predict(img_path):
77
  if pred_class.capitalize() in df["Makanan"].values:
78
  row = df.loc[df["Makanan"] == pred_class.capitalize()]
79
 
80
- # Mengambil informasi gizi
81
- calories = row["Kalori"].values[0]
82
- protein = row["Protein"].values[0]
83
- fat = row["Lemak"].values[0]
84
- carbs = row["Karbohidrat"].values[0]
85
- fiber = row["Serat"].values[0]
86
- sugar = row["Gula"].values[0]
87
- price = row["Harga (Rp)"].values[0]
 
 
 
 
 
 
 
 
 
88
 
89
  return pred_class, calories, protein, fat, carbs, fiber, sugar, price
90
  else:
 
77
  if pred_class.capitalize() in df["Makanan"].values:
78
  row = df.loc[df["Makanan"] == pred_class.capitalize()]
79
 
80
+ # Convert int64 values to native Python data types
81
+ calories = int(row["Kalori"].values[0])
82
+ protein = int(row["Protein"].values[0])
83
+ fat = int(row["Lemak"].values[0])
84
+ carbs = int(row["Karbohidrat"].values[0])
85
+ fiber = int(row["Serat"].values[0])
86
+ sugar = int(row["Gula"].values[0])
87
+ price = int(row["Harga (Rp)"].values[0])
88
+
89
+ # # Mengambil informasi gizi
90
+ # calories = row["Kalori"].values[0]
91
+ # protein = row["Protein"].values[0]
92
+ # fat = row["Lemak"].values[0]
93
+ # carbs = row["Karbohidrat"].values[0]
94
+ # fiber = row["Serat"].values[0]
95
+ # sugar = row["Gula"].values[0]
96
+ # price = row["Harga (Rp)"].values[0]
97
 
98
  return pred_class, calories, protein, fat, carbs, fiber, sugar, price
99
  else: