Spaces:
Running
on
Zero
Running
on
Zero
fix a bug
Browse files
Rodin.py
CHANGED
@@ -17,7 +17,11 @@ def login(email, password):
|
|
17 |
payload['email'] = email
|
18 |
|
19 |
response = requests.post(f"{BASE_URL}/user/login", json=payload)
|
20 |
-
|
|
|
|
|
|
|
|
|
21 |
|
22 |
if 'error' in response_data and response_data['error']:
|
23 |
raise Exception(response_data['error'])
|
@@ -137,8 +141,11 @@ class Generator:
|
|
137 |
print("Error in image preprocessing:", preprocess_response['error'])
|
138 |
else:
|
139 |
if not (prompt and task_uuid):
|
140 |
-
prompt = preprocess_response.get('prompt', 'Default prompt if none returned')
|
141 |
-
|
|
|
|
|
|
|
142 |
|
143 |
return prompt, processed_image
|
144 |
|
|
|
17 |
payload['email'] = email
|
18 |
|
19 |
response = requests.post(f"{BASE_URL}/user/login", json=payload)
|
20 |
+
try:
|
21 |
+
response_data = response.json()
|
22 |
+
except json.JSONDecodeError as e:
|
23 |
+
print(response)
|
24 |
+
raise e
|
25 |
|
26 |
if 'error' in response_data and response_data['error']:
|
27 |
raise Exception(response_data['error'])
|
|
|
141 |
print("Error in image preprocessing:", preprocess_response['error'])
|
142 |
else:
|
143 |
if not (prompt and task_uuid):
|
144 |
+
prompt = preprocess_response.get('prompt', 'Default prompt if none returned')
|
145 |
+
try:
|
146 |
+
processed_image = "data:image/png;base64," + preprocess_response.get('processed_image', None)
|
147 |
+
except Exception as e:
|
148 |
+
print(preprocess_response)
|
149 |
|
150 |
return prompt, processed_image
|
151 |
|