Spaces:
Sleeping
Sleeping
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
- app.py +0 -9
- 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,
|