voxmenthe commited on
Commit
02525f2
·
1 Parent(s): 472f1d2

update README

Browse files
Files changed (1) hide show
  1. README.md +50 -0
README.md CHANGED
@@ -9,6 +9,16 @@ datasets:
9
  metrics:
10
  - accuracy
11
  - f1
 
 
 
 
 
 
 
 
 
 
12
  ---
13
 
14
  # ModernBERT IMDb Sentiment Analysis Model
@@ -69,3 +79,43 @@ print(f"Predicted label: {predicted_label}")
69
 
70
  ### Model Citation
71
  - **Citation**: voxmenthe/modernbert-imdb-sentiment
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
  metrics:
10
  - accuracy
11
  - f1
12
+ title: IMDb Sentiment Analyzer
13
+ emoji: 🤗
14
+ colorFrom: blue
15
+ colorTo: green
16
+ sdk: gradio
17
+ sdk_version: "4.26.0" # Verify this matches your Gradio version in requirements.txt
18
+ app_file: app.py
19
+ pinned: false
20
+ hf_oauth: false
21
+ disable_embedding: false
22
  ---
23
 
24
  # ModernBERT IMDb Sentiment Analysis Model
 
79
 
80
  ### Model Citation
81
  - **Citation**: voxmenthe/modernbert-imdb-sentiment
82
+
83
+ ## IMDb Sentiment Analyzer - Gradio App
84
+
85
+ This repository contains a Gradio application for sentiment analysis of IMDb movie reviews.
86
+ It uses a fine-tuned ModernBERT model hosted on Hugging Face.
87
+
88
+ **Space Link:** [voxmenthe/imdb-sentiment-demo](https://huggingface.co/spaces/voxmenthe/imdb-sentiment-demo)
89
+ **Model Link:** [voxmenthe/modernbert-imdb-sentiment](https://huggingface.co/voxmenthe/modernbert-imdb-sentiment)
90
+
91
+ ## Features
92
+
93
+ * **Text Input**: Analyze custom movie review text.
94
+ * **Random IMDb Sample**: Load a random review from the IMDb test dataset.
95
+ * **Sentiment Prediction**: Classifies sentiment as Positive or Negative.
96
+ * **True Label Display**: Shows the actual IMDb label for loaded samples.
97
+
98
+ ## Setup & Running Locally
99
+
100
+ 1. **Clone the repository (or your Space repository):**
101
+ ```bash
102
+ git clone https://huggingface.co/spaces/voxmenthe/imdb-sentiment-demo
103
+ cd imdb-sentiment-demo
104
+ ```
105
+
106
+ 2. **Install dependencies:**
107
+ Ensure you have Python 3.11+ installed.
108
+ ```bash
109
+ pip install -r requirements.txt
110
+ ```
111
+
112
+ 3. **Run the application:**
113
+ ```bash
114
+ python app.py
115
+ ```
116
+ The application will be available at `http://127.0.0.1:7860`.
117
+
118
+ ## Model Information
119
+
120
+ The sentiment analysis model is a `ModernBERT` architecture fine-tuned on the IMDb dataset. The specific checkpoint used is `mean_epoch5_0.9575acc_0.9575f1.pt` before being uploaded to `voxmenthe/modernbert-imdb-sentiment`.
121
+