manoskary commited on
Commit
2267362
·
verified ·
1 Parent(s): 4977c97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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}")