elnasharomar2 commited on
Commit
7f2524c
·
verified ·
1 Parent(s): d0b8cef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -1,9 +1,14 @@
1
  from transformers import pipeline
2
  import gradio as gr
3
  import os
 
 
4
 
5
  api_key = os.getenv("token")
6
- get_completion = pipeline("ner", model="elnasharomar2/PUNCERT_multi_stages_50_epochs",token = token)
 
 
 
7
 
8
  def merge_tokens(tokens):
9
  merged_tokens = []
 
1
  from transformers import pipeline
2
  import gradio as gr
3
  import os
4
+ from huggingface_hub import login
5
+
6
 
7
  api_key = os.getenv("token")
8
+ login(token = token)
9
+
10
+
11
+ get_completion = pipeline("ner", model="elnasharomar2/PUNCERT_multi_stages_50_epochs")
12
 
13
  def merge_tokens(tokens):
14
  merged_tokens = []