WilliamSotoM commited on
Commit
e231ede
·
1 Parent(s): 13b4a60

Adjusting versions

Browse files
Files changed (3) hide show
  1. README.md +2 -1
  2. app.py +5 -2
  3. requirements.txt +2 -2
README.md CHANGED
@@ -4,7 +4,8 @@ emoji: ⚖️
4
  colorFrom: purple
5
  colorTo: green
6
  sdk: gradio
7
- sdk_version: "5.35.0"
 
8
  app_file: app.py
9
  suggested_hardware: cpu-basic
10
  models:
 
4
  colorFrom: purple
5
  colorTo: green
6
  sdk: gradio
7
+ sdk_version: 5.35.0
8
+ python_version: 3.12
9
  app_file: app.py
10
  suggested_hardware: cpu-basic
11
  models:
app.py CHANGED
@@ -1,6 +1,9 @@
1
  import gradio as gr
2
- from sentence_transformers import CrossEncoder
 
 
3
  import torch
 
4
 
5
  language_mapping = {
6
  'English (eng_Latn)': 'MonoLR_eng_Latn_PR',
@@ -12,7 +15,7 @@ language_mapping = {
12
  }
13
 
14
  print('Loading base model...')
15
- model = CrossEncoder('MoritzLaurer/mDeBERTa-v3-base-xnli-multilingual-nli-2mil7')
16
  print('Base model loaded!')
17
 
18
  model.config.num_labels = 1
 
1
  import gradio as gr
2
+ print('Imported gradio', gr.__version__)
3
+ import sentence_transformers
4
+ print('Imported sentence_transformers', sentence_transformers.__version__)
5
  import torch
6
+ print('Imported torch', torch.__version__)
7
 
8
  language_mapping = {
9
  'English (eng_Latn)': 'MonoLR_eng_Latn_PR',
 
15
  }
16
 
17
  print('Loading base model...')
18
+ model = sentence_transformers.CrossEncoder('MoritzLaurer/mDeBERTa-v3-base-xnli-multilingual-nli-2mil7')
19
  print('Base model loaded!')
20
 
21
  model.config.num_labels = 1
requirements.txt CHANGED
@@ -1,2 +1,2 @@
1
- peft
2
- sentence-transformers
 
1
+ peft==0.10.0
2
+ sentence-transformers==3.2.1