jonathanjordan21 commited on
Commit
ec24a25
1 Parent(s): a9b60a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -24,8 +24,8 @@ sum_llm_chain = tag_chain.chain(llm=local_llm)
24
 
25
  pexels_api_key = os.getenv('pexels_api_key')
26
 
27
- def pred(query):
28
- folder_name, sentences = pexels.generate_videos(query, pexel_api_key, dimension, llm_chain, sum_llm_chain)
29
  utils.combine_videos(folder_name)
30
  return {
31
  'video':folder_name,
@@ -34,7 +34,7 @@ def pred(query):
34
 
35
  with gr.Blocks() as demo:
36
  dimension = gr.Dropdown(
37
- ["Potrait", "Landscape", "Square"], label="Video Dimension", info="Choose dimension"
38
  )
39
  product_name = gr.Textbox(placeholder="Product Name")
40
  captions = gr.Textbox()
 
24
 
25
  pexels_api_key = os.getenv('pexels_api_key')
26
 
27
+ def pred(product_name, orientation):
28
+ folder_name, sentences = pexels.generate_videos(query, pexel_api_key, orientation, llm_chain, sum_llm_chain)
29
  utils.combine_videos(folder_name)
30
  return {
31
  'video':folder_name,
 
34
 
35
  with gr.Blocks() as demo:
36
  dimension = gr.Dropdown(
37
+ ["Potrait", "Landscape", "Square"], label="Video Dimension", info="Choose orientation"
38
  )
39
  product_name = gr.Textbox(placeholder="Product Name")
40
  captions = gr.Textbox()