sreepathi-ravikumar commited on
Commit
4502847
·
verified ·
1 Parent(s): eb632bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -1,7 +1,9 @@
1
- from flask import Flask, redirect
2
  from video import video_generation
 
3
 
4
  app = Flask(__name__)
 
5
 
6
  @app.route('/get_video')
7
  def get_video():
@@ -38,7 +40,8 @@ def get_video():
38
  ]
39
 
40
  video_file = video_generation(text, prompts)
41
- return redirect(video_file)
42
 
43
  if __name__ == '__main__':
44
- app.run(host='0.0.0.0', port=7860)
 
 
1
+ from flask import Flask, send_from_directory
2
  from video import video_generation
3
+ import os
4
 
5
  app = Flask(__name__)
6
+ VIDEO_FOLDER = os.path.abspath(".")
7
 
8
  @app.route('/get_video')
9
  def get_video():
 
40
  ]
41
 
42
  video_file = video_generation(text, prompts)
43
+ return send_from_directory(VIDEO_FOLDER, video_file, mimetype='video/mp4')
44
 
45
  if __name__ == '__main__':
46
+ app.run(host='0.0.0.0', port=7860)
47
+