Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from datasets import load_dataset, Image
+#from datasets import load_dataset, Image
 from huggingface_hub import from_pretrained_keras
 
 x = st.slider('Select a value')
@@ -8,6 +8,7 @@ st.write(x, 'squared is', x * x)
 #loaded_model = keras.saving.load_model("jableable/road_model")
 
 model = from_pretrained_keras("keras-io/ocr-for-captcha")
+model.summary()
 #prediction = model.predict(image)
 #prediction = tf.squeeze(tf.round(prediction))
 #print(f'The image is a {classes[(np.argmax(prediction))]}!')