Commit ·
e231ede
1
Parent(s): 13b4a60
Adjusting versions
Browse files- README.md +2 -1
- app.py +5 -2
- requirements.txt +2 -2
README.md
CHANGED
|
@@ -4,7 +4,8 @@ emoji: ⚖️
|
|
| 4 |
colorFrom: purple
|
| 5 |
colorTo: green
|
| 6 |
sdk: gradio
|
| 7 |
-
sdk_version:
|
|
|
|
| 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 |
-
|
|
|
|
|
|
|
| 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
|