rrg92 commited on
Commit
75128ac
·
verified ·
1 Parent(s): 891634b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -4,8 +4,10 @@ from transformers import AutoTokenizer, AutoModel, AutoImageProcessor
4
  import gradio as gr
5
  import spaces
6
 
7
- model = AutoModel.from_pretrained('neuralmind/bert-base-portuguese-cased')
8
- tokenizer = AutoTokenizer.from_pretrained('neuralmind/bert-base-portuguese-cased', do_lower_case=False)
 
 
9
 
10
  # processor = AutoImageProcessor.from_pretrained("nomic-ai/nomic-embed-vision-v1.5")
11
  # vision_model = AutoModel.from_pretrained("nomic-ai/nomic-embed-vision-v1.5", trust_remote_code=True)
 
4
  import gradio as gr
5
  import spaces
6
 
7
+ # neuralmind/bert-base-portuguese-cased
8
+ ModelName = "papluca/xlm-roberta-base-language-detection"
9
+ model = AutoModel.from_pretrained(ModelName)
10
+ tokenizer = AutoTokenizer.from_pretrained(ModelName, do_lower_case=False)
11
 
12
  # processor = AutoImageProcessor.from_pretrained("nomic-ai/nomic-embed-vision-v1.5")
13
  # vision_model = AutoModel.from_pretrained("nomic-ai/nomic-embed-vision-v1.5", trust_remote_code=True)