Browse Source

Fix (likely) stable diffusion in ai_interface examples

Wieland 2 years ago
parent
commit
fe218051bd
1 changed files with 2 additions and 1 deletions
  1. 2 1
      examples/ai_interface/main.py

+ 2 - 1
examples/ai_interface/main.py

@@ -27,7 +27,8 @@ async def transcribe(e: UploadEventArguments):
 async def generate_image():
 async def generate_image():
     image.source = 'https://dummyimage.com/600x400/ccc/000000.png&text=building+image...'
     image.source = 'https://dummyimage.com/600x400/ccc/000000.png&text=building+image...'
     model = replicate.models.get('stability-ai/stable-diffusion')
     model = replicate.models.get('stability-ai/stable-diffusion')
-    prediction = await io_bound(model.predict, prompt=prompt.value)
+    version = model.versions.get('db21e45d3f7023abc2a46ee38a23973f6dce16bb082a930b0c49861f96d1e5bf')
+    prediction = await io_bound(version.predict, prompt=prompt.value)
     image.source = prediction[0]
     image.source = prediction[0]
 
 
 # User Interface
 # User Interface