Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -25,10 +25,10 @@ def predict_fn(img):
|
|
| 25 |
header_html = """
|
| 26 |
<div style="text-align: center; max-width: 650px; margin: 0 auto;">
|
| 27 |
<img src="imgclass.webp" alt="Header Image" style="max-width: 100%; height: auto; margin: 20px 0;">
|
| 28 |
-
<h1 style="color: #
|
| 29 |
-
<p style="color: #
|
| 30 |
Üdvözöljük képosztályozónkban! Ez az eszköz egy ImageNeten betanított ResNet50 modellt használ
|
| 31 |
-
a képek osztályozására. Tölts fel egy képet, és megmutatjuk az 5 legjobb
|
| 32 |
</p>
|
| 33 |
</div>
|
| 34 |
"""
|
|
@@ -39,11 +39,14 @@ with gr.Blocks() as demo:
|
|
| 39 |
with gr.Row():
|
| 40 |
with gr.Column():
|
| 41 |
input_image = gr.Image(type='pil')
|
| 42 |
-
|
|
|
|
|
|
|
| 43 |
|
| 44 |
with gr.Column():
|
| 45 |
output = gr.Label()
|
| 46 |
|
| 47 |
classify_btn.click(predict_fn, inputs=input_image, outputs=output)
|
|
|
|
| 48 |
|
| 49 |
demo.launch()
|
|
|
|
| 25 |
header_html = """
|
| 26 |
<div style="text-align: center; max-width: 650px; margin: 0 auto;">
|
| 27 |
<img src="imgclass.webp" alt="Header Image" style="max-width: 100%; height: auto; margin: 20px 0;">
|
| 28 |
+
<h1 style="color: #424C5B; font-size: 2.5em;">IT Betyár | Resnet Image Classifier</h1>
|
| 29 |
+
<p style="color: #FFFFFF; font-size: 1.2em;">
|
| 30 |
Üdvözöljük képosztályozónkban! Ez az eszköz egy ImageNeten betanított ResNet50 modellt használ
|
| 31 |
+
a képek osztályozására. Tölts fel egy képet, és megmutatjuk az 5 legjobb predikciót.
|
| 32 |
</p>
|
| 33 |
</div>
|
| 34 |
"""
|
|
|
|
| 39 |
with gr.Row():
|
| 40 |
with gr.Column():
|
| 41 |
input_image = gr.Image(type='pil')
|
| 42 |
+
with gr.Row():
|
| 43 |
+
classify_btn = gr.Button("Mehet")
|
| 44 |
+
clear_btn = gr.Button("Reset")
|
| 45 |
|
| 46 |
with gr.Column():
|
| 47 |
output = gr.Label()
|
| 48 |
|
| 49 |
classify_btn.click(predict_fn, inputs=input_image, outputs=output)
|
| 50 |
+
clear_btn.click(lambda: [None, None], inputs=None, outputs=[input_image, output])
|
| 51 |
|
| 52 |
demo.launch()
|