Upload test_multi_lora.py with huggingface_hub

test_multi_lora.py

@@ -239,12 +239,16 @@ def main():
     summary = json.dumps(results, indent=2, default=str, ensure_ascii=False)
     print(summary)
 
+    sys.stdout.flush()
+
+    # 결과를 파일로 저장
+    result_path = "/tmp/test_results.json"
+    with open(result_path, "w") as f:
+        f.write(summary)
+
     # 결과를 HF repo에 업로드
     try:
         from huggingface_hub import HfApi
-        result_path = "/tmp/test_results.json"
-        with open(result_path, "w") as f:
-            f.write(summary)
         api = HfApi()
         api.upload_file(
             path_or_fileobj=result_path,
@@ -256,10 +260,24 @@ def main():
     except Exception as e:
         print(f"\nFailed to upload results: {e}")
 
-    # 종료하지 않고 대기 (Space가 RUNNING 상태 유지)
-    print("\nTest complete. Keeping container alive...")
-    import signal
-    signal.pause()
+    sys.stdout.flush()
+
+    # 간단한 HTTP 서버로 Space RUNNING 상태 유지 + 결과 제공
+    print("\nTest complete. Starting HTTP server on port 7860...")
+    from http.server import HTTPServer, BaseHTTPRequestHandler
+
+    class ResultHandler(BaseHTTPRequestHandler):
+        def do_GET(self):
+            self.send_response(200)
+            self.send_header("Content-Type", "application/json")
+            self.end_headers()
+            with open(result_path, "rb") as f:
+                self.wfile.write(f.read())
+        def log_message(self, format, *args):
+            pass  # suppress access logs
+
+    server = HTTPServer(("0.0.0.0", 7860), ResultHandler)
+    server.serve_forever()
 
 
 if __name__ == "__main__":