jadechoghari commited on
Commit
181487d
·
verified ·
1 Parent(s): 0d50a2a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -7
app.py CHANGED
@@ -210,13 +210,9 @@ def step_2_generate_video(image):
210
  def step_3_display_3d_model(mesh_file):
211
  return mesh_file
212
 
213
- # set up the example files, we limit to 10
214
- examples = [
215
- "/content/cat.png",
216
- "/content/cat2.png",
217
- "/content/cat3.png",
218
-
219
- ]
220
 
221
  with gr.Blocks() as demo:
222
  with gr.Row():
 
210
  def step_3_display_3d_model(mesh_file):
211
  return mesh_file
212
 
213
+ # set up the example files from assets folder, we limit to 10
214
+ example_folder = "assets"
215
+ examples = [os.path.join(example_folder, f) for f in os.listdir(example_folder) if f.endswith(('.png', '.jpg', '.jpeg'))][:10]
 
 
 
 
216
 
217
  with gr.Blocks() as demo:
218
  with gr.Row():