Dmitry Trifonov commited on
Commit
3e7cff9
·
1 Parent(s): 2e91066

fix demo for Fair 0.15

Browse files
Files changed (1) hide show
  1. fair.py +12 -9
fair.py CHANGED
@@ -8,9 +8,10 @@ logger = logging.getLogger()
8
  import requests
9
  import tempfile
10
 
11
- SERVER_ADRESS="https://faircompute.com:8000/api/v1"
12
- #SERVER_ADRESS="http://localhost:8000/api/v1"
13
- DOCKER_IMAGE="faircompute/stable-diffusion:pytorch-1.13.1-cu116"
 
14
 
15
  class FairApiClient:
16
  def __init__(self, server_address: str):
@@ -62,12 +63,14 @@ class FairApiClient:
62
  return int(response.text)
63
 
64
  def put_job(self, program_id, input_files, output_files):
65
- url = f"{self.server_address}/jobs?program={program_id}"
66
  data = {
 
 
67
  'input_files': input_files,
68
- 'output_files': output_files
 
69
  }
70
-
71
  response = self.put(url=url, data=data)
72
 
73
  return int(response.text)
@@ -93,7 +96,6 @@ class FairApiClient:
93
  response = self.put(url=url, data=None)
94
  return response.text
95
 
96
-
97
  def wait_for_file(self, job_id, path, attempts=10) -> BinaryIO:
98
  headers = {
99
  'Authorization': f'Bearer {self.token}'
@@ -119,8 +121,9 @@ class FairApiClient:
119
 
120
 
121
  def text_to_image(text):
122
- email = os.getenv('FAIRCOMPUTE_EMAIL')
123
- password = os.environ.get('FAIRCOMPUTE_PASSWORD')
 
124
  client = FairApiClient(SERVER_ADRESS)
125
  client.authenticate(email=email, password=password)
126
 
 
8
  import requests
9
  import tempfile
10
 
11
+ SERVER_ADRESS = "https://faircompute.com:8000/api/v1"
12
+ # SERVER_ADRESS = "http://localhost:8000/api/v1"
13
+ DOCKER_IMAGE = "faircompute/stable-diffusion:pytorch-1.13.1-cu116"
14
+
15
 
16
  class FairApiClient:
17
  def __init__(self, server_address: str):
 
63
  return int(response.text)
64
 
65
  def put_job(self, program_id, input_files, output_files):
66
+ url = f"{self.server_address}/jobs"
67
  data = {
68
+ 'type': 'V015',
69
+ 'program': program_id,
70
  'input_files': input_files,
71
+ 'output_files': output_files,
72
+ 'target_node': None,
73
  }
 
74
  response = self.put(url=url, data=data)
75
 
76
  return int(response.text)
 
96
  response = self.put(url=url, data=None)
97
  return response.text
98
 
 
99
  def wait_for_file(self, job_id, path, attempts=10) -> BinaryIO:
100
  headers = {
101
  'Authorization': f'Bearer {self.token}'
 
121
 
122
 
123
  def text_to_image(text):
124
+ # default credentials will work only for local server built in debug mode
125
+ email = os.getenv('FAIRCOMPUTE_EMAIL', "debug-email")
126
+ password = os.environ.get('FAIRCOMPUTE_PASSWORD', "debug-pwd")
127
  client = FairApiClient(SERVER_ADRESS)
128
  client.authenticate(email=email, password=password)
129