zerogpu

app.py

@@ -1,3 +1,4 @@
+import spaces
 import base64
 import io
 import math
@@ -82,6 +83,7 @@ def create_random_word_groups(clues: List[Tuple[List[str], str, str]], target_gr
             groups.append((merged_words, selected_indices))
     return groups
 
+@spaces.GPU(duration=120)
 def group_words(word_list: List[str]) -> List[Group]:
     """
     Groups the given words into 3 to 4 thematic groups.
@@ -130,6 +132,7 @@ def group_words(word_list: List[str]) -> List[Group]:
     print(f"Generated groupings: {generations}")
     return generations.groups
 
+@spaces.GPU(duration=120)
 def generate_clue(group: List[str]) -> Clue:
     """
     Generates a single-word clue for the given group of words.

requirements.txt

@@ -1 +1,2 @@
-outlines
+outlines
+transformers