Hugging Face's logo

Spaces:
k-l-lambda
/
starry
Running

App Files Community
starry
File size: 4,394 Bytes
2b7aae2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
 
# STARRY ML Services - Test Docker Compose
#
# Simplified configuration for testing individual services.
# Uses a single all-in-one image to minimize build time.
#
# Usage:
#   docker compose -f docker-compose.test.yml build
#   docker compose -f docker-compose.test.yml up layout
#   docker compose -f docker-compose.test.yml up ocr

services:
  layout:
    build:
      context: ../../..
      dockerfile: backend/python-services/Dockerfile
      target: all-in-one
    container_name: starry-layout-test
    ports:
      - "12022:12022"
    volumes:
      - /home/camus/data/models/starry/starry-dist:/models/starry-dist:ro
      - /home/camus/work/deep-starry:/app/deep-starry:ro
    environment:
      - CUDA_VISIBLE_DEVICES=0
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    working_dir: /app/deep-starry
    command: >
      python streamPredictor.py
      /models/starry-dist/20221125-scorelayout-1121-residue-u-d4-w64-d4-w64
      -p 12022 -dv cuda -m layout

  mask:
    build:
      context: ../../..
      dockerfile: backend/python-services/Dockerfile
      target: all-in-one
    container_name: starry-mask-test
    ports:
      - "12024:12024"
    volumes:
      - /home/camus/data/models/starry/starry-dist:/models/starry-dist:ro
      - /home/camus/work/deep-starry:/app/deep-starry:ro
    environment:
      - CUDA_VISIBLE_DEVICES=0
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    working_dir: /app/deep-starry
    command: >
      python streamPredictor.py
      /models/starry-dist/20210918-scorewidgets.mask-unet-5-32
      -p 12024 -dv cuda -m mask

  semantic:
    build:
      context: ../../..
      dockerfile: backend/python-services/Dockerfile
      target: all-in-one
    container_name: starry-semantic-test
    ports:
      - "12025:12025"
    volumes:
      - /home/camus/data/models/starry/starry-dist:/models/starry-dist:ro
      - /home/camus/work/deep-starry:/app/deep-starry:ro
    environment:
      - CUDA_VISIBLE_DEVICES=0
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    working_dir: /app/deep-starry
    command: >
      python streamPredictor.py
      /models/starry-dist/202302-semanticCluster
      -p 12025 -dv cuda -m semanticCluster

  loc:
    build:
      context: ../../..
      dockerfile: backend/python-services/Dockerfile
      target: all-in-one
    container_name: starry-loc-test
    ports:
      - "12026:12026"
    volumes:
      - /home/camus/data/models/starry/ocr-dist:/models/ocr-dist:ro
      - /home/camus/work/starry-ocr:/app/starry-ocr:ro
    environment:
      - CUDA_VISIBLE_DEVICES=0
      - LOG_DIR=/tmp/starry-logs
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    working_dir: /app/starry-ocr
    command: >
      python locPredictor.py
      -w /models/ocr-dist/DB_gc_loc/v6/model_epoch_88_minibatch_15300
      -p 12026 -dv cuda

  ocr:
    build:
      context: ../../..
      dockerfile: backend/python-services/Dockerfile
      target: all-in-one
    container_name: starry-ocr-test
    ports:
      - "12027:12027"
    volumes:
      - /home/camus/data/models/starry/ocr-dist:/models/ocr-dist:ro
      - /home/camus/work/starry-ocr:/app/starry-ocr:ro
    environment:
      - CUDA_VISIBLE_DEVICES=-1
      - TF_USE_LEGACY_KERAS=1
      - LOG_DIR=/tmp/starry-logs
    working_dir: /app/starry-ocr
    command: >
      python ocrPredictor.py
      /models/ocr-dist/ocr.yaml
      -p 12027

  brackets:
    build:
      context: ../../..
      dockerfile: backend/python-services/Dockerfile
      target: all-in-one
    container_name: starry-brackets-test
    ports:
      - "12028:12028"
    volumes:
      - /home/camus/data/models/starry/ocr-dist:/models/ocr-dist:ro
      - /home/camus/work/starry-ocr:/app/starry-ocr:ro
    environment:
      - CUDA_VISIBLE_DEVICES=-1
      - TF_USE_LEGACY_KERAS=1
      - LOG_DIR=/tmp/starry-logs
    working_dir: /app/starry-ocr
    command: >
      python bracketsPredictor.py
      /models/ocr-dist/brackets.yaml
      -p 12028