Spaces:

AMfeta99
/

Object_Evolution_Generator

Running

AMfeta99 commited on May 22

Commit

0aed780

verified ·

1 Parent(s): 5ace3a2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -109,7 +109,7 @@ class TextToImageTool(Tool):
         device = "cuda" if torch.cuda.is_available() else "cpu"
         print(f"Using device: {device}")
         self.pipe = DiffusionPipeline.from_pretrained(
-            "black-forest-labs/FLUX.1-schnell",
             torch_dtype=dtype
         ).to(device)

         device = "cuda" if torch.cuda.is_available() else "cpu"
         print(f"Using device: {device}")
         self.pipe = DiffusionPipeline.from_pretrained(
+            "aiyouthalliance/Free-Image-Generation-CC0",
             torch_dtype=dtype
         ).to(device)