Hugging Face's logo

Spaces:
jrmd
/
News_article_classifier_based_on_BERT_model
Sleeping

App Files Community
jrmd commited on
Commit
dd0fd10
·
1 Parent(s): 60ef55c

Interface improvements

Browse files
Files changed (3) hide show
  1. W&B Chart 20_05_2025 00_42_07.png +0 -3
  2. W&B Chart 20_05_2025 00_42_20.png +0 -3
  3. demo.py +13 -6
W&B Chart 20_05_2025 00_42_07.png DELETED

Git LFS Details

  • SHA256: 5fe03054854608a6bc58209d79d4aadf130a28a01cf51ab0436dbae739a0ad10
  • Pointer size: 131 Bytes
  • Size of remote file: 471 kB
W&B Chart 20_05_2025 00_42_20.png DELETED

Git LFS Details

  • SHA256: 68f926f7a172e41a8e52521f5bf5b109bd33c7690611d06be300d66d4f2f1310
  • Pointer size: 131 Bytes
  • Size of remote file: 307 kB
demo.py CHANGED
@@ -63,10 +63,17 @@ def find_category(
63
  return y_pred
64
 
65
 
66
- demo = gr.Interface(
67
- fn=find_category,
68
- inputs=["text"],
69
- outputs=["text"],
70
- )
 
 
 
 
 
 
 
71
 
72
- demo.launch()
 
63
  return y_pred
64
 
65
 
66
+ with gr.Blocks() as demo:
67
+ gr.Markdown(
68
+ "<h1 style='text-align: center;'>CUSTOM MODEL BASED ON BERT BASE TO CLASSIFY NEWS ARTICLES</h1>"
69
+ )
70
+ gr.Markdown(
71
+ "<h2 style='text-align: center;'>Model loss during training and eval time</h2>"
72
+ )
73
+
74
+ with gr.Row():
75
+ gr.Image(value="wandb_chart_train.png", label="Training Loss")
76
+ gr.Image(value="wandb_chart_eval.png", label="Eval Loss")
77
+ gr.Interface(fn=find_category, inputs=["text"], outputs=["text"], live=False)
78
 
79
+ demo.launch(share=True)