Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -156,7 +156,7 @@ class SimpleNotaGenTool(Tool):
|
|
| 156 |
# 3. Zero GPU is used efficiently (shorter GPU allocation times)
|
| 157 |
|
| 158 |
try:
|
| 159 |
-
notagen_tool = SimpleNotaGenTool(device=device)
|
| 160 |
logger.info("✅ NotaGen tool initialized successfully!")
|
| 161 |
except Exception as e:
|
| 162 |
logger.error(f"❌ Failed to initialize NotaGen tool: {e}")
|
|
|
|
| 156 |
# 3. Zero GPU is used efficiently (shorter GPU allocation times)
|
| 157 |
|
| 158 |
try:
|
| 159 |
+
notagen_tool = SimpleNotaGenTool(device=device, model_id="ElectricAlexis/NotaGen")
|
| 160 |
logger.info("✅ NotaGen tool initialized successfully!")
|
| 161 |
except Exception as e:
|
| 162 |
logger.error(f"❌ Failed to initialize NotaGen tool: {e}")
|