balthou commited on
Commit
3a04f9a
·
1 Parent(s): a94f6bf

remove image generatio dependancy

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -3,7 +3,7 @@ from interactive_pipe.headless.pipeline import HeadlessPipeline
3
  from interactive_pipe.graphical.gradio_gui import InteractivePipeGradio
4
  from interactive_pipe import interactive
5
  from interactive_pipe.data_objects.image import Image
6
- from interactive_pipe.thirdparty.images_openai_api import ImageFromPrompt
7
  from pathlib import Path
8
  import cv2
9
 
@@ -45,11 +45,12 @@ TRACK_DICT = {
45
  }
46
 
47
  for item_name, element in TRACK_DICT.items():
48
- TRACK_DICT[item_name][IMAGE] = ImageFromPrompt.generate_image(
49
- PROMPT_STYLE + element[PROMPT] + PROMPT_EXTRA,
50
- img_folder/(item_name+".png"),
51
- size=(512, 512)
52
- )
 
53
  ICONS = [it[ICON] for key, it in TRACK_DICT.items()]
54
 
55
 
 
3
  from interactive_pipe.graphical.gradio_gui import InteractivePipeGradio
4
  from interactive_pipe import interactive
5
  from interactive_pipe.data_objects.image import Image
6
+ # from interactive_pipe.thirdparty.images_openai_api import ImageFromPrompt
7
  from pathlib import Path
8
  import cv2
9
 
 
45
  }
46
 
47
  for item_name, element in TRACK_DICT.items():
48
+ # TRACK_DICT[item_name][IMAGE] = ImageFromPrompt.generate_image(
49
+ # PROMPT_STYLE + element[PROMPT] + PROMPT_EXTRA,
50
+ # img_folder/(item_name+".png"),
51
+ # size=(512, 512)
52
+ # )
53
+ TRACK_DICT[item_name][IMAGE] = img_folder/(item_name+".png")
54
  ICONS = [it[ICON] for key, it in TRACK_DICT.items()]
55
 
56