Spaces:

salomonsky
/

flux3

Running

salomonsky commited on Oct 19

Commit

0d349dc

•

1 Parent(s): e80b447

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -30,7 +30,7 @@ if not HF_TOKEN_UPSCALER:
 def get_upscale_finegrain(prompt, img_path, upscale_factor):
     try:
-        upscale_client = InferenceClient("finegrain/finegrain-image-enhancer", hf_token=HF_TOKEN_UPSCALER)
         result = upscale_client.predict(input_image=handle_file(img_path), prompt=prompt, upscale_factor=upscale_factor)
         return result[1] if isinstance(result, list) and len(result) > 1 else None
     except Exception as e:

 def get_upscale_finegrain(prompt, img_path, upscale_factor):
     try:
+        upscale_client = InferenceClient("fal/AuraSR-v2", hf_token=HF_TOKEN_UPSCALER)
         result = upscale_client.predict(input_image=handle_file(img_path), prompt=prompt, upscale_factor=upscale_factor)
         return result[1] if isinstance(result, list) and len(result) > 1 else None
     except Exception as e: