Update app.py

app.py

@@ -3,30 +3,31 @@ import torch
 import ai_thingy
 
 
-def respond(message, history, system_message, max_tokens, temperature, top_k):
+# Hugging Face Spaces can use CPU unless you've
+# explicitly enabled a GPU.
+ai_thingy.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
 
-    # --- AOBANCORP™ AUTO-AWAKEN ---
-    if ai_thingy.current_model is None:
-        print("[SYSTEM] Loading Aoban 1.1A-Refined...")
+print("[SYSTEM] Initializing Aoban 1.1A-Refined...")
+print(f"[SYSTEM] Device: {ai_thingy.device}")
 
-        ai_thingy.device = torch.device("cpu")
+ai_thingy.initialize_or_retrain(
+    initial_train=True,
+    use_live_data=False,
+    epochs=0
+)
 
-        ai_thingy.initialize_or_retrain(
-            initial_train=True,
-            use_live_data=False,
-            epochs=0
-        )
+print("[SYSTEM] Aoban 1.1A-Refined is ready.")
 
-        print("[SYSTEM] Aoban 1.1A-Refined loaded.")
 
-    # --- PROMPT ---
+def respond(message, history, system_message, max_tokens, temperature, top_k):
+
     full_prompt = (
         f"{system_message}\n"
         f"User: {message}\n"
         f"Assistant:"
     )
 
-    # --- GENERATION ---
     response = ai_thingy.generate_text(
         model=ai_thingy.current_model,
         tokenizer=ai_thingy.current_tokenizer,
@@ -48,22 +49,25 @@ demo = gr.ChatInterface(
     additional_inputs=[
         gr.Textbox(
             value="You are a helpful but not really helpful AobanCorp™ assistant.",
-            label="An Aoban label"
+            label="Aoban System Prompt"
         ),
         gr.Slider(
-            10, 100,
+            minimum=10,
+            maximum=100,
             value=50,
             step=1,
             label="Response Length"
         ),
         gr.Slider(
-            0.1, 2.0,
+            minimum=0.1,
+            maximum=2.0,
             value=1.0,
             step=0.1,
-            label="Creativity (Temp)"
+            label="Creativity (Temperature)"
         ),
         gr.Slider(
-            1, 100,
+            minimum=1,
+            maximum=100,
             value=40,
             step=1,
             label="Focus (Top-K)"