Spaces:
Runtime error
Runtime error
✨ update default model to V2
Browse filesSigned-off-by: peter szemraj <peterszemraj@gmail.com>
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")
|