lfcc commited on
Commit
3ad0902
1 Parent(s): 2a0848a

streamlit error fix

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +2 -1
app.py CHANGED
@@ -18,7 +18,7 @@ questions = json.load(file)
18
  model = AutoModelForTokenClassification.from_pretrained("models/trigger/ACE_base_google_pt")
19
  tokenizer = AutoTokenizer.from_pretrained("models/trigger/ACE_base_google_pt", model_max_length=512)
20
  tagger = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy='first') #aggregation_strategy='max'
21
-
22
  qa_model2_path = "models/argument/bert-base-portuguese-squad2-ace-impossible1.4"
23
  qa_model_path = "models/argument/bert-base-portuguese-squad-ace-pt-pt"
24
  model_qa = pipeline("question-answering", model=qa_model_path)
 
18
  model = AutoModelForTokenClassification.from_pretrained("models/trigger/ACE_base_google_pt")
19
  tokenizer = AutoTokenizer.from_pretrained("models/trigger/ACE_base_google_pt", model_max_length=512)
20
  tagger = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy='first') #aggregation_strategy='max'
21
+
22
  qa_model2_path = "models/argument/bert-base-portuguese-squad2-ace-impossible1.4"
23
  qa_model_path = "models/argument/bert-base-portuguese-squad-ace-pt-pt"
24
  model_qa = pipeline("question-answering", model=qa_model_path)
requirements.txt CHANGED
@@ -3,4 +3,5 @@ torch
3
  transformers
4
  tqdm
5
  numpy
6
- pysbd
 
 
3
  transformers
4
  tqdm
5
  numpy
6
+ pysbd
7
+ altair==4.1.0