pushpikaLiyanagama's picture
Update app.py
a7b659a verified
raw
history blame
2.43 kB
import joblib
import pandas as pd
import gradio as gr
# Load the scaler and models
scaler = joblib.load("models/scaler.joblib")
models = {
"processing": joblib.load("models/svm_model_processing.joblib"),
"perception": joblib.load("models/svm_model_perception.joblib"),
"input": joblib.load("models/svm_model_input.joblib"),
"understanding": joblib.load("models/svm_model_understanding.joblib")
}
def predict(course_overview, reading_file, abstract_materiale, concrete_material, visual_materials,
self_assessment, exercises_submit, quiz_submitted, playing, paused, unstarted, buffering):
try:
input_data = {
"course overview": [course_overview],
"reading file": [reading_file],
"abstract materiale": [abstract_materiale],
"concrete material": [concrete_material],
"visual materials": [visual_materials],
"self-assessment": [self_assessment],
"exercises submit": [exercises_submit],
"quiz submitted": [quiz_submitted],
"playing": [playing],
"paused": [paused],
"unstarted": [unstarted],
"buffering": [buffering]
}
input_df = pd.DataFrame(input_data)
input_scaled = scaler.transform(input_df)
predictions = {}
for target, model in models.items():
pred = model.predict(input_scaled)
predictions[target] = pred[0] # Return as is, without converting to int
return predictions
except Exception as e:
return {"error": str(e)}
# Define Gradio interface using the latest syntax
iface = gr.Interface(
fn=predict,
inputs=[
gr.Number(label="Course Overview"),
gr.Number(label="Reading File"),
gr.Number(label="Abstract Materiale"),
gr.Number(label="Concrete Material"),
gr.Number(label="Visual Materials"),
gr.Number(label="Self Assessment"),
gr.Number(label="Exercises Submit"),
gr.Number(label="Quiz Submitted"),
gr.Number(label="Playing"),
gr.Number(label="Paused"),
gr.Number(label="Unstarted"),
gr.Number(label="Buffering")
],
outputs=gr.JSON(),
title="SVM Multi-Target Prediction",
description="Enter the feature values to get predictions for processing, perception, input, and understanding."
)
if __name__ == "__main__":
iface.launch()