Hugging Face's logo

Spaces:
salomonsky
/
flux3
Running

App Files Community
salomonsky commited on
Commit
0d349dc
•
1 Parent(s): e80b447

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ if not HF_TOKEN_UPSCALER:
30
 
31
  def get_upscale_finegrain(prompt, img_path, upscale_factor):
32
  try:
33
- upscale_client = InferenceClient("finegrain/finegrain-image-enhancer", hf_token=HF_TOKEN_UPSCALER)
34
  result = upscale_client.predict(input_image=handle_file(img_path), prompt=prompt, upscale_factor=upscale_factor)
35
  return result[1] if isinstance(result, list) and len(result) > 1 else None
36
  except Exception as e:
 
30
 
31
  def get_upscale_finegrain(prompt, img_path, upscale_factor):
32
  try:
33
+ upscale_client = InferenceClient("fal/AuraSR-v2", hf_token=HF_TOKEN_UPSCALER)
34
  result = upscale_client.predict(input_image=handle_file(img_path), prompt=prompt, upscale_factor=upscale_factor)
35
  return result[1] if isinstance(result, list) and len(result) > 1 else None
36
  except Exception as e: