Update app.py
Browse files
app.py
CHANGED
|
@@ -11,42 +11,34 @@ def predict(COUNT, COUNT_CV, STRENGTH, CSP, U_PERCENT, THIN, THICK, NEPS, IPI):
|
|
| 11 |
prediction = model.predict(features)[0][0]
|
| 12 |
return round(float(prediction), 2)
|
| 13 |
|
| 14 |
-
#
|
| 15 |
-
slider_COUNT = gr.Slider(minimum=7, maximum=50, step=0.5, label="COUNT")
|
| 16 |
-
slider_COUNT_CV = gr.Slider(minimum=0.48, maximum=1.87, step=0.1, label="COUNT_CV")
|
| 17 |
-
slider_STRENGTH = gr.Slider(minimum=62.71, maximum=384.2, step=0.1, label="STRENGTH")
|
| 18 |
-
|
| 19 |
-
slider_CSP = gr.Slider(minimum=1500, maximum=4755, step=1, label="CSP")
|
| 20 |
-
slider_U_PERCENT = gr.Slider(minimum=6.38, maximum=12.12, step=0.1, label="U_PERCENT")
|
| 21 |
-
slider_THIN = gr.Slider(minimum=0.0, maximum=19, step=0.1, label="THIN")
|
| 22 |
-
|
| 23 |
-
slider_THICK = gr.Slider(minimum=2.0, maximum=150, step=1, label="THICK")
|
| 24 |
-
slider_NEPS = gr.Slider(minimum=1, maximum=494, step=1, label="NEPS")
|
| 25 |
-
slider_IPI = gr.Slider(minimum=6.0, maximum=646, step=1, label="IPI")
|
| 26 |
-
|
| 27 |
with gr.Blocks() as demo:
|
| 28 |
gr.Markdown("## 🧵 Textile Mixing Predictor (TensorFlow Model)")
|
| 29 |
gr.Markdown("Adjust the sliders for each feature and get the predicted output.")
|
| 30 |
|
| 31 |
with gr.Row():
|
| 32 |
with gr.Column():
|
| 33 |
-
|
| 34 |
-
|
| 35 |
-
|
| 36 |
with gr.Column():
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
with gr.Column():
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
|
| 44 |
|
| 45 |
output = gr.Number(label="Predicted Output")
|
| 46 |
|
| 47 |
-
|
| 48 |
|
| 49 |
-
|
| 50 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
|
| 52 |
demo.launch()
|
|
|
|
| 11 |
prediction = model.predict(features)[0][0]
|
| 12 |
return round(float(prediction), 2)
|
| 13 |
|
| 14 |
+
# Gradio app with custom sliders
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
with gr.Blocks() as demo:
|
| 16 |
gr.Markdown("## 🧵 Textile Mixing Predictor (TensorFlow Model)")
|
| 17 |
gr.Markdown("Adjust the sliders for each feature and get the predicted output.")
|
| 18 |
|
| 19 |
with gr.Row():
|
| 20 |
with gr.Column():
|
| 21 |
+
COUNT = gr.Slider(minimum=7, maximum=50, step=0.5, label="COUNT")
|
| 22 |
+
COUNT_CV = gr.Slider(minimum=0.48, maximum=1.87, step=0.1, label="COUNT_CV")
|
| 23 |
+
STRENGTH = gr.Slider(minimum=62.71, maximum=384.2, step=0.1, label="STRENGTH")
|
| 24 |
with gr.Column():
|
| 25 |
+
CSP = gr.Slider(minimum=1500, maximum=4755, step=1, label="CSP")
|
| 26 |
+
U_PERCENT = gr.Slider(minimum=6.38, maximum=12.12, step=0.1, label="U_PERCENT")
|
| 27 |
+
THIN = gr.Slider(minimum=0.0, maximum=19, step=0.1, label="THIN")
|
| 28 |
with gr.Column():
|
| 29 |
+
THICK = gr.Slider(minimum=2.0, maximum=150, step=1, label="THICK")
|
| 30 |
+
NEPS = gr.Slider(minimum=1, maximum=494, step=1, label="NEPS")
|
| 31 |
+
IPI = gr.Slider(minimum=6.0, maximum=646, step=1, label="IPI")
|
| 32 |
|
| 33 |
output = gr.Number(label="Predicted Output")
|
| 34 |
|
| 35 |
+
predict_button = gr.Button("Predict")
|
| 36 |
|
| 37 |
+
# Bind inputs to function
|
| 38 |
+
predict_button.click(
|
| 39 |
+
fn=predict,
|
| 40 |
+
inputs=[COUNT, COUNT_CV, STRENGTH, CSP, U_PERCENT, THIN, THICK, NEPS, IPI],
|
| 41 |
+
outputs=output
|
| 42 |
+
)
|
| 43 |
|
| 44 |
demo.launch()
|