Spaces:

ruslanmv
/

TextToVideo-Flux

Paused

ruslanmv commited on Feb 2

Commit

d354a33

1 Parent(s): c0408e6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -74,6 +74,7 @@ DESCRIPTION = (
 )
 TITLE = "Video Story Generator with Audio by using FLUX, distilbart, and GTTS."
 # Load Tokenizer and Model for Text Summarization
 def load_text_summarization_model():
     """Load the tokenizer and model for text summarization on CPU."""
     print("Loading text summarization model...")

 )
 TITLE = "Video Story Generator with Audio by using FLUX, distilbart, and GTTS."
 # Load Tokenizer and Model for Text Summarization
+@spaces.GPU()
 def load_text_summarization_model():
     """Load the tokenizer and model for text summarization on CPU."""
     print("Loading text summarization model...")