bibibi12345 commited on
Commit
365b2e0
·
1 Parent(s): df7abb4

fixed resolution

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -266,12 +266,14 @@ def generate():
266
  is_text_to_video = 'text-to-video' in model_endpoint
267
  is_image_to_video = 'image-to-video' in model_endpoint
268
 
 
 
 
 
 
269
  if is_text_to_image:
270
  # Image generation (text-to-image)
271
- if 'image_size' in data:
272
- fal_arguments['image_size'] = data['image_size']
273
- if 'num_images' in data:
274
- fal_arguments['num_images'] = data['num_images']
275
  elif is_text_to_video:
276
  # Video generation (text-to-video)
277
  for k in ['aspect_ratio', 'resolution', 'duration', 'camera_fixed']:
 
266
  is_text_to_video = 'text-to-video' in model_endpoint
267
  is_image_to_video = 'image-to-video' in model_endpoint
268
 
269
+ if 'image_size' in data:
270
+ fal_arguments['image_size'] = data['image_size']
271
+ if 'num_images' in data:
272
+ fal_arguments['num_images'] = data['num_images']
273
+
274
  if is_text_to_image:
275
  # Image generation (text-to-image)
276
+ pass
 
 
 
277
  elif is_text_to_video:
278
  # Video generation (text-to-video)
279
  for k in ['aspect_ratio', 'resolution', 'duration', 'camera_fixed']: