kovacsvi commited on
Commit
6d86fd5
·
1 Parent(s): 8027e9b

removed spaces before gradio downgrade

Browse files
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
- #@spaces.GPU(duration=30)
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