seawolf2357 commited on
Commit
dcafb3e
·
verified ·
1 Parent(s): 3cd1683

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -7
app.py CHANGED
@@ -1,11 +1,8 @@
1
  import os
2
- import subprocess
3
- import sys
4
  import io
5
  import gradio as gr
6
  import numpy as np
7
  import random
8
- import spaces
9
  import torch
10
  from diffusers import Flux2Pipeline, Flux2Transformer2DModel
11
  from diffusers import BitsAndBytesConfig as DiffBitsAndBytesConfig
@@ -15,7 +12,7 @@ import json
15
  import base64
16
  from huggingface_hub import InferenceClient
17
 
18
- subprocess.check_call([sys.executable, "-m", "pip", "install", "spaces==0.43.0"])
19
 
20
  dtype = torch.bfloat16
21
  device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -162,9 +159,9 @@ def update_dimensions_from_image(image_list):
162
 
163
  return new_width, new_height
164
 
165
- # ⚠️ spaces.GPU 데코레이터는 파라미터 개수 제한이 있음
166
- # 모든 파라미터를 하나의 dict로 묶어서 전달
167
- @spaces.GPU
168
  def generate_image(generation_args):
169
  """
170
  generation_args = {
 
1
  import os
 
 
2
  import io
3
  import gradio as gr
4
  import numpy as np
5
  import random
 
6
  import torch
7
  from diffusers import Flux2Pipeline, Flux2Transformer2DModel
8
  from diffusers import BitsAndBytesConfig as DiffBitsAndBytesConfig
 
12
  import base64
13
  from huggingface_hub import InferenceClient
14
 
15
+ # spaces 라이브러리 제거 - ZeroGPU가 자동으로 GPU 할당
16
 
17
  dtype = torch.bfloat16
18
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
159
 
160
  return new_width, new_height
161
 
162
+ # ⚠️ @spaces.GPU 데코레이터 제거
163
+ # spaces 0.43.0에서 데코레이터가 작동하지 않음
164
+ # GPU는 ZeroGPU가 자동으로 할당함
165
  def generate_image(generation_args):
166
  """
167
  generation_args = {