road_project / app.py
jableable's picture
Update app.py
14b8d21 verified
raw
history blame
553 Bytes
import streamlit as st
from datasets import load_dataset, Image
from huggingface_hub import from_pretrained_keras
x = st.slider('Select a value')
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")
#prediction = model.predict(image)
#prediction = tf.squeeze(tf.round(prediction))
#print(f'The image is a {classes[(np.argmax(prediction))]}!')
#dataset = load_dataset("beans", split="train")
#loaded_img = dataset[0]["image"]
#print(loaded_img)