szili2011's picture
Create app.py
6e0d345 verified
raw
history blame contribute delete
814 Bytes
import gradio as gr
from tensorflow.keras.models import load_model
import numpy as np
# Load the saved model
model = load_model('time_prediction_model.h5')
def predict_time(features):
# Assuming features is a list of input values
input_data = np.array(features).reshape(1, -1)
hour_prediction, minute_prediction = model.predict(input_data)
# Convert predictions to readable format
predicted_hour = np.argmax(hour_prediction)
predicted_minute = np.argmax(minute_prediction)
return f"{predicted_hour:02}:{predicted_minute:02}"
# Create a Gradio interface
interface = gr.Interface(
fn=predict_time,
inputs=[gr.inputs.Textbox(lines=2, placeholder="Enter input features")],
outputs="text",
title="Time Prediction AI"
)
# Launch the interface
interface.launch()