official.ghost.logic commited on
Commit
425645b
Β·
1 Parent(s): be976cc

Add @spaces.GPU decorator to AI generation functions

Browse files

- Decorate create_character_ui with @spaces.GPU
- Enables ZeroGPU acceleration for AI model calls
- Satisfies HF Spaces GPU detection requirement

Files changed (2) hide show
  1. app.py +0 -9
  2. src/ui/tabs/character_create_tab.py +2 -0
app.py CHANGED
@@ -13,12 +13,6 @@ sys.path.insert(0, str(Path(__file__).parent))
13
 
14
  from src.ui import launch_ui
15
 
16
- # Dummy GPU function to satisfy HF Spaces ZeroGPU requirement
17
- @spaces.GPU
18
- def initialize_app():
19
- """Initialize app - decorated for HF Spaces compatibility"""
20
- return True
21
-
22
  if __name__ == "__main__":
23
  print("""
24
  ╔═══════════════════════════════════════════════╗
@@ -30,9 +24,6 @@ if __name__ == "__main__":
30
  """)
31
 
32
  try:
33
- # Call GPU-decorated function
34
- initialize_app()
35
- # Launch the UI
36
  launch_ui()
37
  except KeyboardInterrupt:
38
  print("\n\nπŸ‘‹ Shutting down gracefully...")
 
13
 
14
  from src.ui import launch_ui
15
 
 
 
 
 
 
 
16
  if __name__ == "__main__":
17
  print("""
18
  ╔═══════════════════════════════════════════════╗
 
24
  """)
25
 
26
  try:
 
 
 
27
  launch_ui()
28
  except KeyboardInterrupt:
29
  print("\n\nπŸ‘‹ Shutting down gracefully...")
src/ui/tabs/character_create_tab.py CHANGED
@@ -5,6 +5,7 @@ Character Create Tab for D'n'D Campaign Manager
5
  import gradio as gr
6
  from typing import Tuple
7
  import traceback
 
8
  from src.agents.character_agent import CharacterAgent
9
  from src.models.character import DnDRace, DnDClass, Alignment
10
  from src.utils.validators import get_available_races, get_available_classes
@@ -94,6 +95,7 @@ Generate only the name:"""
94
  except Exception as e:
95
  return f"Error: {str(e)}"
96
 
 
97
  def create_character_ui(
98
  self,
99
  name: str,
 
5
  import gradio as gr
6
  from typing import Tuple
7
  import traceback
8
+ import spaces
9
  from src.agents.character_agent import CharacterAgent
10
  from src.models.character import DnDRace, DnDClass, Alignment
11
  from src.utils.validators import get_available_races, get_available_classes
 
95
  except Exception as e:
96
  return f"Error: {str(e)}"
97
 
98
+ @spaces.GPU
99
  def create_character_ui(
100
  self,
101
  name: str,