Hugging Face's logo

Spaces:
CreatorIQ-org
/
ls_be_T5_base
Paused

App Files Community
b2u commited on
Commit
20eab11
·
1 Parent(s): 603d158

resolving dependencies

Browse files
Files changed (2) hide show
  1. model.py +1 -2
  2. requirements-base.txt +4 -1
model.py CHANGED
@@ -7,8 +7,7 @@ import json
7
  import torch
8
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
9
  from peft import get_peft_model, LoraConfig, TaskType, PeftModel
10
- from label_studio_ml.model import LabelStudioMLBase
11
- from label_studio_ml.response import ModelResponse
12
 
13
  logger = logging.getLogger(__name__)
14
 
 
7
  import torch
8
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
9
  from peft import get_peft_model, LoraConfig, TaskType, PeftModel
10
+ from label_studio_ml.model import LabelStudioMLBase, ModelResponse
 
11
 
12
  logger = logging.getLogger(__name__)
13
 
requirements-base.txt CHANGED
@@ -1,2 +1,5 @@
1
  gunicorn==22.0.0
2
- label-studio-ml @ git+https://github.com/HumanSignal/label-studio-ml-backend.git
 
 
 
 
1
  gunicorn==22.0.0
2
+ label-studio-ml-backend==1.0.9
3
+ transformers
4
+ torch
5
+ peft