Spaces:
Sleeping
Sleeping
import gradio as gr | |
import numpy as np | |
import tensorflow as tf | |
# Assuming your model is loaded here | |
model = tf.keras.models.load_model("pain_analysis.h5") | |
def predict(image): | |
image = np.array(image) / 255.0 # Normalize the image | |
# Make predictions using your model here | |
# result = model.predict(image) | |
return result | |
# Change this line to use gr.Image directly | |
iface = gr.Interface( | |
fn=predict, | |
inputs=gr.Image(type="numpy"), # Use gr.Image directly | |
outputs="label" # Specify the output type as needed | |
) | |
iface.launch() | |