From 68559eb18033b047425ee81440ddcafbe02906f6 Mon Sep 17 00:00:00 2001
From: Haoming <73768377+Haoming02@users.noreply.github.com>
Date: Sat, 7 Sep 2024 11:45:51 +0800
Subject: [PATCH] space requirement (#1721)

---
 modules_forge/forge_space.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/modules_forge/forge_space.py b/modules_forge/forge_space.py
index ca73795b..62bbc353 100644
--- a/modules_forge/forge_space.py
+++ b/modules_forge/forge_space.py
@@ -132,6 +132,12 @@ class ForgeSpace:
         os.makedirs(self.hf_path, exist_ok=True)
 
         if self.repo_id is None:
+            requirements_filename = os.path.abspath(os.path.realpath(os.path.join(self.root_path, 'requirements.txt')))
+
+            if os.path.exists(requirements_filename):
+                from modules.launch_utils import run_pip
+                run_pip(f'install -r "{requirements_filename}"', desc=f"space requirements for [{self.title}]")
+
             return self.refresh_gradio()
 
         downloaded = snapshot_download(