Falln87 commited on
Commit
0feaeea
1 Parent(s): 7f51054

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ inference_api = inference_api.InferenceApi(token=inference_api_token)
21
 
22
  # Set up the Diffusers library
23
  diffusers_device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
24
- diffusers_model = diffusers.DDPMPipeline.from_pretrained("stabilityai/stable-diffusion-2")
25
 
26
  def recognize_speech():
27
  with sr.Microphone() as source:
 
21
 
22
  # Set up the Diffusers library
23
  diffusers_device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
24
+ diffusers_model = diffusers.DDPMPipeline.from_pretrained("ByteDance/SDXL-Lightning")
25
 
26
  def recognize_speech():
27
  with sr.Microphone() as source: