Deploy Myco from CI

game/engine.py

@@ -1,9 +1,26 @@
 """Core Myco gameplay — LLM is the primary game engine."""
 
 try:
-    import spaces  # must be imported before any CUDA/transformers code on HF Spaces
+    import spaces
+
+    @spaces.GPU
+    def _run_pipeline(pipe, messages):
+        return pipe(
+            messages,
+            max_new_tokens=80,
+            temperature=0.85,
+            do_sample=True,
+            return_full_text=False,
+        )
 except ImportError:
-    pass  # not running on HF Spaces, ignore
+    def _run_pipeline(pipe, messages):
+        return pipe(
+            messages,
+            max_new_tokens=80,
+            temperature=0.85,
+            do_sample=True,
+            return_full_text=False,
+        )
 
 import os
 import random
@@ -150,13 +167,7 @@ def _llm(prompt: str, context: dict | None = None) -> str | None:
     ]
 
     try:
-        outputs = pipe(
-            messages,
-            max_new_tokens=80,
-            temperature=0.85,
-            do_sample=True,
-            return_full_text=False,
-        )
+        outputs = _run_pipeline(pipe, messages)
         print("========== MYCO OUTPUT ==========")
         print(outputs)
         print("=================================")
@@ -170,7 +181,7 @@ def _llm(prompt: str, context: dict | None = None) -> str | None:
         print(f"[Myco] Inference error: {exc}")
         traceback.print_exc()
         return None
-
+        
 
 def _llm_with_history(history: list, user_message: str, context: dict) -> str | None:
     """Call Gemma with full conversation history for the chat interface."""
@@ -208,16 +219,10 @@ def _llm_with_history(history: list, user_message: str, context: dict) -> str |
     messages.append({"role": "user", "content": user_message})
 
     try:
-        outputs = pipe(
-            messages,
-            max_new_tokens=80,
-            temperature=0.85,
-            do_sample=True,
-            return_full_text=False,
-        )
-        print("========== MYCO CHAT OUTPUT ==========")
+        outputs = _run_pipeline(pipe, messages)
+        print("========== MYCO OUTPUT ==========")
         print(outputs)
-        print("======================================")
+        print("=================================")
         generated = outputs[0].get("generated_text", "")
         if isinstance(generated, list):
             last = generated[-1]
@@ -225,7 +230,7 @@ def _llm_with_history(history: list, user_message: str, context: dict) -> str |
         return str(generated).strip() or None
     except Exception as exc:
         import traceback
-        print(f"[Myco] Chat inference error: {exc}")
+        print(f"[Myco] Inference error: {exc}")
         traceback.print_exc()
         return None