dev-bjoern commited on
Commit
8037a7c
·
verified ·
1 Parent(s): c02bf22

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -12
app.py CHANGED
@@ -131,20 +131,22 @@ class InfinigenManager:
131
  self._run_command(f"git clone https://github.com/princeton-vl/infinigen.git {self.base_dir}",
132
  "Fehler beim Klonen von Infinigen", timeout=30)
133
 
134
- if not self._is_infinigen_installed():
135
- logger.info("Installiere Infinigen...")
136
- self._run_command([
137
- self.blender_python, "-m", "pip", "install", "-e", f"{self.base_dir}[terrain,vis]", "--no-deps", "--user"
138
- ], "Fehler bei der Installation von Infinigen", timeout=60)
139
-
140
- def _is_infinigen_installed(self):
141
- """Prüft, ob Infinigen bereits installiert ist."""
142
- if not self.blender_python:
143
- return False
 
144
  try:
145
- self._run_command([self.blender_python, "-c", "import infinigen"], timeout=10)
146
  return True
147
- except Exception:
 
148
  return False
149
 
150
  @spaces.GPU
 
131
  self._run_command(f"git clone https://github.com/princeton-vl/infinigen.git {self.base_dir}",
132
  "Fehler beim Klonen von Infinigen", timeout=30)
133
 
134
+ # Installiere Infinigen mit Abhängigkeiten
135
+ self._run_command([
136
+ self.blender_python, "-m", "pip", "install", "-e", f"{self.base_dir}[terrain,vis]", "--user"
137
+ ], "Fehler bei der Installation von Infinigen", timeout=60)
138
+
139
+ # Überprüfe, ob das datagen-Modul verfügbar ist
140
+ if not self._is_datagen_available():
141
+ raise RuntimeError("Das 'datagen'-Modul von Infinigen ist nicht verfügbar. Überprüfe die Installation.")
142
+
143
+ def _is_datagen_available(self):
144
+ """Prüft, ob das 'datagen'-Modul von Infinigen verfügbar ist."""
145
  try:
146
+ self._run_command([self.blender_python, "-c", "import infinigen.datagen"], timeout=10)
147
  return True
148
+ except Exception as e:
149
+ logger.error(f"Fehler beim Import von 'infinigen.datagen': {e}")
150
  return False
151
 
152
  @spaces.GPU