Spaces:
Running
on
Zero
Running
on
Zero
update
Browse files
demo/binary_classifier_demo.py
CHANGED
|
@@ -8,7 +8,6 @@ import spaces
|
|
| 8 |
from model_utils import load_model, classify_text
|
| 9 |
from binoculars_utils import initialize_binoculars, compute_scores
|
| 10 |
|
| 11 |
-
DEVICE = torch.device('cuda')
|
| 12 |
MINIMUM_TOKENS = 200
|
| 13 |
|
| 14 |
SAMPLE_TEXT = """Привет! Я хотел бы рассказать вам о своём опыте путешествия по Петербургу. Невероятный город с богатой историей и красивой архитектурой. Особенно запомнился Эрмитаж с его огромной коллекцией произведений искусства. Также понравилась прогулка по каналам города, где можно увидеть множество старинных мостов и зданий."""
|
|
|
|
| 8 |
from model_utils import load_model, classify_text
|
| 9 |
from binoculars_utils import initialize_binoculars, compute_scores
|
| 10 |
|
|
|
|
| 11 |
MINIMUM_TOKENS = 200
|
| 12 |
|
| 13 |
SAMPLE_TEXT = """Привет! Я хотел бы рассказать вам о своём опыте путешествия по Петербургу. Невероятный город с богатой историей и красивой архитектурой. Особенно запомнился Эрмитаж с его огромной коллекцией произведений искусства. Также понравилась прогулка по каналам города, где можно увидеть множество старинных мостов и зданий."""
|