Spaces:
Running
Running
✨ update default model to V2
Browse filesSigned-off-by: peter szemraj <[email protected]>
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
|
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")
|