CarolineM5 commited on
Commit
8b336ad
·
verified ·
1 Parent(s): 4a822f2

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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/model_LR"
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