Spaces:
Running
on
Zero
Running
on
Zero
fix a bug
Browse files
Rodin.py
CHANGED
@@ -125,12 +125,15 @@ def load_image(img_path):
|
|
125 |
class Generator:
|
126 |
def __init__(self, user_id, password) -> None:
|
127 |
_, self.token = login(user_id, password)
|
|
|
|
|
128 |
self.task_uuid = None
|
129 |
|
130 |
def preprocess(self, prompt, image_path, cache_image_base64, task_uuid=""):
|
131 |
if cache_image_base64 and prompt and task_uuid != "":
|
132 |
return prompt, cache_image_base64
|
133 |
print("Preprocessing image...")
|
|
|
134 |
|
135 |
image_file = load_image(image_path)
|
136 |
if prompt and task_uuid:
|
|
|
125 |
class Generator:
|
126 |
def __init__(self, user_id, password) -> None:
|
127 |
_, self.token = login(user_id, password)
|
128 |
+
self.user_id = user_id
|
129 |
+
self.password = password
|
130 |
self.task_uuid = None
|
131 |
|
132 |
def preprocess(self, prompt, image_path, cache_image_base64, task_uuid=""):
|
133 |
if cache_image_base64 and prompt and task_uuid != "":
|
134 |
return prompt, cache_image_base64
|
135 |
print("Preprocessing image...")
|
136 |
+
_, self.token = login(self.user_id, self.password)
|
137 |
|
138 |
image_file = load_image(image_path)
|
139 |
if prompt and task_uuid:
|