air3 / app.py
howlbz's picture
Update app.py
d13438a
raw
history blame
693 Bytes
import gradio as gr
import numpy as np
from PIL import Image
import requests
import hopsworks
import joblib
project = hopsworks.login(api_key_value="0rdWXlLgEd3mkGOg.iRZ7TtAkWGPlJHNQcAEph6Qbokoaq7QTBRI9ckwWUki8tIYGyBvrKhJvtLoUOGQ4")
fs = project.get_feature_store()
mr = project.get_model_registry()
model = mr.get_model("xgboost_model", version=1)
model_dir = model.download()
model = joblib.load(model_dir + "/model.pkl")
def forecast():
x = 0
#res = model.predict(x)
return 10
demo = gr.Interface(
fn=titanic,
title="Air Quality Prediction",
description="Get aqi value",
allow_flagging="never",
outputs=gr.Textbox(label="Result: "))
demo.launch()