sanjanatule commited on
Commit
5f7693c
·
1 Parent(s): 9fdde30

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -0
app.py CHANGED
@@ -43,6 +43,17 @@ import litmodelclass
43
 
44
 
45
  # gradio
 
 
 
 
 
 
 
 
 
 
 
46
  title = "Yolo3 trained on PASCAL_VOC with GradCAM"
47
  description = "Gradio interface to infer on Yolo3 model, and get GradCAM results"
48
 
@@ -185,5 +196,13 @@ with gr.Blocks() as demo:
185
  outputs=img_output,
186
  fn=yolo3_inference, cache_examples=False)
187
 
 
 
 
 
 
 
 
 
188
  if __name__ == "__main__":
189
  demo.launch()
 
43
 
44
 
45
  # gradio
46
+
47
+ model_stats = """
48
+ ### YoloV3 Model Implementation & Training Details
49
+ Github Link: https://github.com/santule/ERA/tree/main/S13
50
+ #### Model Performance:
51
+ 1. **Validation Loss: 6.05**
52
+ 2. **Class accuracy: 82.4%**
53
+ 3. **No obj accuracy: 98.05%**
54
+ 4. **Obj accuracy: 72.3%**
55
+ """
56
+
57
  title = "Yolo3 trained on PASCAL_VOC with GradCAM"
58
  description = "Gradio interface to infer on Yolo3 model, and get GradCAM results"
59
 
 
196
  outputs=img_output,
197
  fn=yolo3_inference, cache_examples=False)
198
 
199
+
200
+ with gr.Row():
201
+ with gr.Box():
202
+ with gr.Row():
203
+ with gr.Column():
204
+ with gr.Box():
205
+ gr.Markdown(model_stats)
206
+
207
  if __name__ == "__main__":
208
  demo.launch()