fixing bugs
Browse files
app.py
CHANGED
|
@@ -66,9 +66,6 @@ def full_pipeline(image, scale_factor, model_type, style_prompt):
|
|
| 66 |
sr_np = np.asarray(sr_jax)
|
| 67 |
sr_pil = Image.fromarray(sr_np)
|
| 68 |
|
| 69 |
-
if device == "cpu":
|
| 70 |
-
return sr_pil, None, None
|
| 71 |
-
|
| 72 |
# 2. Style Transfer (PyTorch)
|
| 73 |
prompt = f"BAS-RELIEF {style_prompt}, intricate carving, marble texture"
|
| 74 |
bas_relief = pipe(
|
|
|
|
| 66 |
sr_np = np.asarray(sr_jax)
|
| 67 |
sr_pil = Image.fromarray(sr_np)
|
| 68 |
|
|
|
|
|
|
|
|
|
|
| 69 |
# 2. Style Transfer (PyTorch)
|
| 70 |
prompt = f"BAS-RELIEF {style_prompt}, intricate carving, marble texture"
|
| 71 |
bas_relief = pipe(
|