Spaces:
Sleeping
Sleeping
Upload 2 files
Browse files- app.py +1 -1
- requirements.txt +2 -1
app.py
CHANGED
|
@@ -26,7 +26,7 @@ def gradio_generate(fibers_map: Image.Image, rings_map: Image.Image) -> Image.Im
|
|
| 26 |
fibers_map = fibers_map.convert("RGB")
|
| 27 |
rings_map = rings_map.convert("RGB")
|
| 28 |
|
| 29 |
-
model_id = "CarolineM5/InstructPix2Pix_WithoutPrompt
|
| 30 |
|
| 31 |
result_img = inference(model_id, device, rings_map, fibers_map)
|
| 32 |
|
|
|
|
| 26 |
fibers_map = fibers_map.convert("RGB")
|
| 27 |
rings_map = rings_map.convert("RGB")
|
| 28 |
|
| 29 |
+
model_id = "CarolineM5/InstructPix2Pix_WithoutPrompt"
|
| 30 |
|
| 31 |
result_img = inference(model_id, device, rings_map, fibers_map)
|
| 32 |
|
requirements.txt
CHANGED
|
@@ -3,4 +3,5 @@ torch>=2.0.0
|
|
| 3 |
transformers>=4.36.0
|
| 4 |
diffusers>=0.25.0
|
| 5 |
Pillow
|
| 6 |
-
numpy
|
|
|
|
|
|
| 3 |
transformers>=4.36.0
|
| 4 |
diffusers>=0.25.0
|
| 5 |
Pillow
|
| 6 |
+
numpy
|
| 7 |
+
accelerate
|