Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ import warnings
|
|
4 |
import os
|
5 |
import gradio as gr
|
6 |
import numpy as np
|
7 |
-
|
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 |
-
|
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,
|