Spaces:
Running
Running
kovacsvi
commited on
Commit
·
6d86fd5
1
Parent(s):
8027e9b
removed spaces before gradio downgrade
Browse files- interfaces/cap.py +0 -1
- interfaces/cap_minor_media.py +2 -2
- requirements.txt +0 -1
interfaces/cap.py
CHANGED
@@ -1,5 +1,4 @@
|
|
1 |
import gradio as gr
|
2 |
-
import spaces
|
3 |
|
4 |
import os
|
5 |
import time
|
|
|
1 |
import gradio as gr
|
|
|
2 |
|
3 |
import os
|
4 |
import time
|
interfaces/cap_minor_media.py
CHANGED
@@ -1,5 +1,4 @@
|
|
1 |
import gradio as gr
|
2 |
-
import spaces
|
3 |
|
4 |
import os
|
5 |
import torch
|
@@ -55,13 +54,14 @@ def check_huggingface_path(checkpoint_path: str):
|
|
55 |
except:
|
56 |
return False
|
57 |
|
|
|
58 |
def build_huggingface_path(language: str, domain: str, hierarchical: bool):
|
59 |
if hierarchical:
|
60 |
return ("poltextlab/xlm-roberta-large-pooled-cap-media", "poltextlab/xlm-roberta-large-pooled-cap-minor-v3")
|
61 |
else:
|
62 |
return "poltextlab/xlm-roberta-large-pooled-cap-media-minor"
|
63 |
|
64 |
-
|
65 |
def predict(text, major_model_id, minor_model_id, tokenizer_id, HF_TOKEN=None):
|
66 |
device = torch.device("cpu")
|
67 |
|
|
|
1 |
import gradio as gr
|
|
|
2 |
|
3 |
import os
|
4 |
import torch
|
|
|
54 |
except:
|
55 |
return False
|
56 |
|
57 |
+
|
58 |
def build_huggingface_path(language: str, domain: str, hierarchical: bool):
|
59 |
if hierarchical:
|
60 |
return ("poltextlab/xlm-roberta-large-pooled-cap-media", "poltextlab/xlm-roberta-large-pooled-cap-minor-v3")
|
61 |
else:
|
62 |
return "poltextlab/xlm-roberta-large-pooled-cap-media-minor"
|
63 |
|
64 |
+
|
65 |
def predict(text, major_model_id, minor_model_id, tokenizer_id, HF_TOKEN=None):
|
66 |
device = torch.device("cpu")
|
67 |
|
requirements.txt
CHANGED
@@ -8,4 +8,3 @@ huspacy
|
|
8 |
numpy==1.26.4
|
9 |
safetensors
|
10 |
huggingface_hub
|
11 |
-
spaces
|
|
|
8 |
numpy==1.26.4
|
9 |
safetensors
|
10 |
huggingface_hub
|
|