jonathanjordan21 commited on
Commit
b110be1
1 Parent(s): 5f5a645

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import torch
3
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, AutoModelForSeq2SeqLM
4
 
5
  from components import caption_chain, tag_chain
6
- from components import pexels
7
  import os
8
 
9
  model = AutoModelForSeq2SeqLM.from_pretrained("declare-lab/flan-alpaca-gpt4-xl")
@@ -25,7 +25,7 @@ pexels_api_key = os.getenv('pexels_api_key')
25
 
26
  def pred():
27
  folder_name, sentences = pexels.generate_videos("Bluetooth Earphone", pexel_api_key, 1920, 1080)
28
- pexels.combine_videos(folder_name)
29
  return {
30
  'video':folder_name,
31
  'captions':sentences.join("\n")
 
3
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, AutoModelForSeq2SeqLM
4
 
5
  from components import caption_chain, tag_chain
6
+ from components import pexels, utils
7
  import os
8
 
9
  model = AutoModelForSeq2SeqLM.from_pretrained("declare-lab/flan-alpaca-gpt4-xl")
 
25
 
26
  def pred():
27
  folder_name, sentences = pexels.generate_videos("Bluetooth Earphone", pexel_api_key, 1920, 1080)
28
+ utils.combine_videos(folder_name)
29
  return {
30
  'video':folder_name,
31
  'captions':sentences.join("\n")