Nymbo commited on
Commit
c3e02d6
1 Parent(s): 060a7c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ from diffusers import AutoencoderKL, EulerDiscreteScheduler, UNet2DConditionMode
5
  from esrgan_model import UpscalerESRGAN
6
  import gradio as gr
7
  from huggingface_hub import hf_hub_download
8
- import spaces
9
  import torch
10
  import torch.nn as nn
11
  from torchvision.io import read_image
@@ -92,7 +92,7 @@ torch.cuda.empty_cache()
92
 
93
 
94
  # Define image generation function
95
- @spaces.GPU(duration=10)
96
  @torch.no_grad()
97
  def generate_image(
98
  input_image, seed: int = 42, guidance_scale: float = 2.0, num_inference_steps: int = 50, is_upscale: bool = False
 
5
  from esrgan_model import UpscalerESRGAN
6
  import gradio as gr
7
  from huggingface_hub import hf_hub_download
8
+ # import spaces
9
  import torch
10
  import torch.nn as nn
11
  from torchvision.io import read_image
 
92
 
93
 
94
  # Define image generation function
95
+ # @spaces.GPU(duration=10)
96
  @torch.no_grad()
97
  def generate_image(
98
  input_image, seed: int = 42, guidance_scale: float = 2.0, num_inference_steps: int = 50, is_upscale: bool = False