Ginidu2003 commited on
Commit
25346d7
·
verified ·
1 Parent(s): fd0b3b1

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +8 -0
src/streamlit_app.py CHANGED
@@ -8,6 +8,14 @@ from nltk.stem import WordNetLemmatizer
8
  import re
9
  import string
10
 
 
 
 
 
 
 
 
 
11
  # ====================== PREPROCESSING (Same as Task 2) ======================
12
 
13
  # ====================== LOAD FINE-TUNED MODEL ======================
 
8
  import re
9
  import string
10
 
11
+ import os
12
+ from huggingface_hub import login
13
+
14
+ hf_token = os.getenv("HF_TOKEN")
15
+ if hf_token:
16
+ login(hf_token)
17
+
18
+
19
  # ====================== PREPROCESSING (Same as Task 2) ======================
20
 
21
  # ====================== LOAD FINE-TUNED MODEL ======================