malvin noel commited on
Commit
fe4c8dd
Β·
1 Parent(s): ea30417

add space gpu

Browse files
Files changed (2) hide show
  1. app.py +2 -1
  2. requirements.txt +0 -0
app.py CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
2
  import os
3
  import shutil
4
  from typing import List, Optional
 
5
 
6
  from scripts.generate_scripts import generate_script, generate_title, generate_description
7
  from scripts.generate_voice import generate_voice
@@ -29,7 +30,7 @@ def safe_copy(src: str, dst: str) -> str:
29
  # ──────────────────────────────────────────────────────────────────────────────
30
  # Core processing pipeline
31
  # ──────────────────────────────────────────────────────────────────────────────
32
-
33
  def process_video(
34
  context: str,
35
  instruction: str,
 
2
  import os
3
  import shutil
4
  from typing import List, Optional
5
+ import spaces
6
 
7
  from scripts.generate_scripts import generate_script, generate_title, generate_description
8
  from scripts.generate_voice import generate_voice
 
30
  # ──────────────────────────────────────────────────────────────────────────────
31
  # Core processing pipeline
32
  # ──────────────────────────────────────────────────────────────────────────────
33
+ @spaces.GPU()
34
  def process_video(
35
  context: str,
36
  instruction: str,
requirements.txt CHANGED
Binary files a/requirements.txt and b/requirements.txt differ