Nymbo commited on
Commit
43a2073
1 Parent(s): 44edad9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,7 +4,7 @@ import warnings
4
  import os
5
  import gradio as gr
6
  import numpy as np
7
- # import spaces
8
  import torch
9
  from diffusers import FluxControlNetModel
10
  from diffusers.pipelines import FluxControlNetPipeline
@@ -81,7 +81,7 @@ def process_input(input_image, upscale_factor, **kwargs):
81
  return input_image.resize((w, h)), w_original, h_original, was_resized
82
 
83
 
84
- # @spaces.GPU#(duration=42)
85
  def infer(
86
  seed,
87
  randomize_seed,
 
4
  import os
5
  import gradio as gr
6
  import numpy as np
7
+ import spaces
8
  import torch
9
  from diffusers import FluxControlNetModel
10
  from diffusers.pipelines import FluxControlNetPipeline
 
81
  return input_image.resize((w, h)), w_original, h_original, was_resized
82
 
83
 
84
+ @spaces.GPU#(duration=42)
85
  def infer(
86
  seed,
87
  randomize_seed,