jableable commited on
Commit
dbdd9ed
1 Parent(s): 0ad39d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -8,13 +8,13 @@ st.write(x, 'squared is', x * x)
8
  #loaded_model = keras.saving.load_model("best_model.keras")
9
 
10
  model = from_pretrained_keras("best_model.keras")
11
- prediction = model.predict(image)
12
- prediction = tf.squeeze(tf.round(prediction))
13
- print(f'The image is a {classes[(np.argmax(prediction))]}!')
14
 
15
 
16
- dataset = load_dataset("beans", split="train")
17
 
18
- loaded_img = dataset[0]["image"]
19
- print(loaded_img)
20
 
 
8
  #loaded_model = keras.saving.load_model("best_model.keras")
9
 
10
  model = from_pretrained_keras("best_model.keras")
11
+ #prediction = model.predict(image)
12
+ #prediction = tf.squeeze(tf.round(prediction))
13
+ #print(f'The image is a {classes[(np.argmax(prediction))]}!')
14
 
15
 
16
+ #dataset = load_dataset("beans", split="train")
17
 
18
+ #loaded_img = dataset[0]["image"]
19
+ #print(loaded_img)
20