Delje commited on
Commit
0e83f70
·
verified ·
1 Parent(s): 3cc3b07

Create requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -29
requirements.txt CHANGED
@@ -1,29 +1,2 @@
1
- from fastapi import FastAPI, UploadFile
2
- import ffmpeg
3
- from transformers import pipeline
4
-
5
- app = FastAPI()
6
-
7
- # Load Hugging Face model (example: text-to-video)
8
- model = pipeline("text-to-video-generation", model="damo-vilab/modelscope-text-to-video-synthesis")
9
-
10
- @app.post("/generate_video/")
11
- async def generate_video(prompt: str):
12
- # Generate video from text
13
- result = model(prompt)
14
- output_file = "generated_video.mp4"
15
- with open(output_file, "wb") as f:
16
- f.write(result["video"])
17
-
18
- return {"message": "Video generated successfully!", "file_path": output_file}
19
-
20
- @app.post("/upload_video/")
21
- async def upload_video(file: UploadFile):
22
- input_file = f"uploaded_{file.filename}"
23
- with open(input_file, "wb") as f:
24
- f.write(file.file.read())
25
-
26
- # Example: Trim video using FFmpeg
27
- trimmed_file = "trimmed_video.mp4"
28
- ffmpeg.input(input_file).output(trimmed_file, ss=0, t=10).run()
29
- return {"message": "Video uploaded and trimmed!", "file_path": trimmed_file}
 
1
+ fastapi
2
+ uvicorn[standard]