Spaces:

CohereLabs
/

aya_expanse

Running on T4

shivalikasingh commited on Jan 12

Commit

16a7386

verified ·

1 Parent(s): bd124f5

revert to flux-schnell

Files changed (1) hide show

app.py CHANGED Viewed

@@ -113,7 +113,7 @@ def replicate_api_inference(input_prompt):
         "safety_tolerance": 1,
         "num_inference_steps": 4
     }
-    image = replicate.run("black-forest-labs/flux-schnell-lora",input=input_params)
     image = Image.open(image[0])
     return image

         "safety_tolerance": 1,
         "num_inference_steps": 4
     }
+    image = replicate.run("black-forest-labs/flux-schnell",input=input_params)
     image = Image.open(image[0])
     return image