From f73b2505d70c89823dd1ac1c1c04d9d19c0cf54d Mon Sep 17 00:00:00 2001
From: hodanov <1031hoda@gmail.com>
Date: Tue, 12 Sep 2023 20:42:40 +0900
Subject: [PATCH] Repair some bugs.

---
 sdcli/txt2img.py        | 2 +-
 setup_files/__main__.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/sdcli/txt2img.py b/sdcli/txt2img.py
index f97379a..4574477 100644
--- a/sdcli/txt2img.py
+++ b/sdcli/txt2img.py
@@ -32,7 +32,7 @@ def main(
         if seed == -1:
             seed_generated = util.generate_seed()
         start_time = time.time()
-        images = stub.app.run_inference.call(
+        images = stub.app.run_inference.remote(
             prompt=prompt,
             n_prompt=n_prompt,
             height=height,
diff --git a/setup_files/__main__.py b/setup_files/__main__.py
index 7500a76..e5fb5f5 100644
--- a/setup_files/__main__.py
+++ b/setup_files/__main__.py
@@ -7,7 +7,7 @@ from txt2img import StableDiffusion, new_stable_diffusion
 @stub.function(gpu="A10G")
 def main():
     sd = new_stable_diffusion()
-    print(isinstance(sd, StableDiffusion))
+    print(f"Deploy '{sd.__class__.__name__}'.")
 
 
 if __name__ == "__main__":