Oscar Wang commited on
Commit
aa2c783
·
verified ·
1 Parent(s): 9bc335b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -6,6 +6,7 @@ import torch
6
  model_options = {
7
  "GoalZero/aidetection-ada-v0.2": "GoalZero/aidetection-ada-v0.2",
8
  "GoalZero/aidetection-ada-v0.1": "GoalZero/aidetection-ada-v0.1"
 
9
  }
10
 
11
  # Initialize global variables for model and tokenizer
 
6
  model_options = {
7
  "GoalZero/aidetection-ada-v0.2": "GoalZero/aidetection-ada-v0.2",
8
  "GoalZero/aidetection-ada-v0.1": "GoalZero/aidetection-ada-v0.1"
9
+ "GoalZero/babbage-mini-v0.1": "GoalZero/babbage-mini-v0.1"
10
  }
11
 
12
  # Initialize global variables for model and tokenizer