pszemraj commited on
Commit
f81af70
1 Parent(s): ed735aa

✨ update default model to V2

Browse files

Signed-off-by: peter szemraj <peterszemraj@gmail.com>

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -100,7 +100,7 @@ def get_parser():
100
  "--model",
101
  required=False,
102
  type=str,
103
- default="postbot/distilgpt2-emailgen",
104
  help="Pass an different huggingface model tag to use a custom model",
105
  )
106
 
@@ -119,7 +119,7 @@ Hello,
119
 
120
  Following up on last week's bubblegum shipment, I"""
121
 
122
- available_models = ['postbot/distilgpt2-emailgen', 'postbot/distilgpt2-emailgen-V2', 'postbot/gpt2-medium-emailgen']
123
 
124
  if __name__ == "__main__":
125
  logging.info("\n\n\nStarting new instance of app.py")
 
100
  "--model",
101
  required=False,
102
  type=str,
103
+ default="postbot/distilgpt2-emailgen-V2",
104
  help="Pass an different huggingface model tag to use a custom model",
105
  )
106
 
 
119
 
120
  Following up on last week's bubblegum shipment, I"""
121
 
122
+ available_models = ['postbot/distilgpt2-emailgen-V2', 'postbot/distilgpt2-emailgen', 'postbot/gpt2-medium-emailgen']
123
 
124
  if __name__ == "__main__":
125
  logging.info("\n\n\nStarting new instance of app.py")