my-api-name2 / app.py
Abdul234's picture
Create app.py
fa71c8a verified
raw
history blame contribute delete
719 Bytes
import gradio as gr
import joblib
# Load the trained model
model = joblib.load("model.pkl") # Ensure the filename matches exactly
# Define a function to make predictions
def predict(input_data):
try:
input_data = [float(i) for i in input_data.split(',')] # Convert input string to a list of numbers
prediction = model.predict([input_data]) # Make prediction
return str(prediction[0]) # Convert output to string
except Exception as e:
return f"Error: {str(e)}"
# Create Gradio API
interface = gr.Interface(
fn=predict,
inputs=gr.Textbox(placeholder="Enter comma-separated numbers (e.g., 1.5, 2.3, 3.1)"),
outputs="text"
)
# Launch the API
interface.launch()