dstars commited on
Commit
74cd6e9
·
verified ·
1 Parent(s): f136d79

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -35,7 +35,8 @@ def main():
35
 
36
  # Set the device to CPU explicitly
37
  agent = ConversationalAgent(model_path=MODEL_PATH,
38
- outputs_dir=OUTPUT_PATH) # Pass the device to the agent
 
39
 
40
  theme = CustomTheme()
41
 
 
35
 
36
  # Set the device to CPU explicitly
37
  agent = ConversationalAgent(model_path=MODEL_PATH,
38
+ outputs_dir=OUTPUT_PATH,
39
+ device='cpu') # Pass the device to the agent
40
 
41
  theme = CustomTheme()
42