appinitdev commited on
Commit
cf4e354
verified
1 Parent(s): faabd9b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -10
app.py CHANGED
@@ -5,29 +5,24 @@ import torch
5
  import cv2
6
  import numpy as np
7
  from PIL import Image
8
- from diffusers import StableDiffusionPipeline, DDIMScheduler
9
- # NUEVO: Importamos la funci贸n para descargar
10
- from huggingface_hub import hf_hub_download
11
 
12
  # --- FORZAR RUTA PARA QUE ENCUENTRE EL M脫DULO ---
13
  sys.path.append(os.path.dirname(os.path.abspath(__file__)))
14
  from ip_adapter.ip_adapter_faceid import IPAdapterFaceID
15
- # -----------------------------------------------
16
 
17
  from insightface.app import FaceAnalysis
18
 
19
  # Configuraci贸n
20
- device = "cpu" # Si cambias a GPU en settings, c谩mbialo a "cuda"
21
  model_id = "runwayml/stable-diffusion-v1-5"
22
 
23
- # NUEVO: Descarga autom谩tica del modelo
24
- # Esto guardar谩 el archivo en la cach茅 de Hugging Face y devolver谩 la ruta local
25
  print("Descargando/Verificando el modelo IP-Adapter...")
26
  ip_ckpt = hf_hub_download(repo_id="h94/IP-Adapter-FaceID", filename="ip-adapter-faceid_sd15.bin")
27
- print(f"Modelo cargado desde: {ip_ckpt}")
28
 
29
  # Carga de modelos
30
- # NOTA: Si est谩s en CPU, usa ['CPUExecutionProvider']
31
  app = FaceAnalysis(name="buffalo_l", providers=['CPUExecutionProvider'])
32
  app.prepare(ctx_id=0, det_size=(640, 640))
33
 
@@ -47,5 +42,6 @@ def generate(image, prompt):
47
  )
48
  return results[0]
49
 
 
50
  demo = gr.Interface(fn=generate, inputs=[gr.Image(type="pil"), gr.Textbox()], outputs="image")
51
- demo.launch()
 
5
  import cv2
6
  import numpy as np
7
  from PIL import Image
8
+ from diffusers import StableDiffusionPipeline
9
+ from huggingface_hub import hf_hub_download
 
10
 
11
  # --- FORZAR RUTA PARA QUE ENCUENTRE EL M脫DULO ---
12
  sys.path.append(os.path.dirname(os.path.abspath(__file__)))
13
  from ip_adapter.ip_adapter_faceid import IPAdapterFaceID
 
14
 
15
  from insightface.app import FaceAnalysis
16
 
17
  # Configuraci贸n
18
+ device = "cpu"
19
  model_id = "runwayml/stable-diffusion-v1-5"
20
 
21
+ # Descarga autom谩tica del modelo
 
22
  print("Descargando/Verificando el modelo IP-Adapter...")
23
  ip_ckpt = hf_hub_download(repo_id="h94/IP-Adapter-FaceID", filename="ip-adapter-faceid_sd15.bin")
 
24
 
25
  # Carga de modelos
 
26
  app = FaceAnalysis(name="buffalo_l", providers=['CPUExecutionProvider'])
27
  app.prepare(ctx_id=0, det_size=(640, 640))
28
 
 
42
  )
43
  return results[0]
44
 
45
+ # Lanzar con opciones optimizadas para estabilidad
46
  demo = gr.Interface(fn=generate, inputs=[gr.Image(type="pil"), gr.Textbox()], outputs="image")
47
+ demo.launch(ssr_mode=False)