itbetyar commited on
Commit
7c0f1c0
·
verified ·
1 Parent(s): 356fe98

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
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: #2c3e50; font-size: 2.5em;">IT Betyár Resnet- Image Classifier</h1>
29
- <p style="color: #34495e; 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 előrejelzést.
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
- classify_btn = gr.Button("Osztályozás")
 
 
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()