Spaces:
Sleeping
Sleeping
import pandas as pd | |
import joblib | |
import gradio as gr | |
from datetime import datetime | |
# Muat model | |
model = joblib.load('random_forest_model.pkl') | |
# Fungsi untuk memprediksi prioritas tugas | |
def predict_task_duration(duration, deadline): | |
start_date = datetime(2024, 10, 20) | |
deadline_days = (pd.to_datetime(deadline) - start_date).days | |
input_data = pd.DataFrame([[duration, deadline_days]], columns=['duration', 'deadline_days']) | |
# Prediksi prioritas | |
priority = model.predict(input_data) | |
return f'Prioritas Tugas: {priority[0]}' | |
# Buat antarmuka Gradio | |
iface = gr.Interface( | |
fn=predict_task_duration, | |
inputs=[ | |
gr.inputs.Slider(minimum=1, maximum=10, default=5, label="Durasi Tugas (dalam jam)"), | |
gr.inputs.Textbox(placeholder="Masukkan tanggal deadline (YYYY-MM-DD)", label="Tanggal Deadline") | |
], | |
outputs="text", | |
title="Prediksi Prioritas Tugas", | |
description="Masukkan durasi dan tanggal deadline tugas untuk memprediksi prioritasnya." | |
) | |
# Jalankan antarmuka | |
iface.launch() | |