Mike Gabriel
commited on
Commit
·
002443a
1
Parent(s):
e0a30a3
Add prompt
Browse files- .gitignore +2 -1
- main.py +2 -2
.gitignore
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
.vscode
|
| 2 |
__pycache__
|
| 3 |
.git
|
| 4 |
-
.wolf*
|
|
|
|
|
|
| 1 |
.vscode
|
| 2 |
__pycache__
|
| 3 |
.git
|
| 4 |
+
.wolf*
|
| 5 |
+
.env
|
main.py
CHANGED
|
@@ -79,5 +79,5 @@ else:
|
|
| 79 |
}, 'model.pth')
|
| 80 |
print("Training complete!")
|
| 81 |
print("Generating response...\n")
|
| 82 |
-
resp = run_model(m, 256, '
|
| 83 |
-
print("Response
|
|
|
|
| 79 |
}, 'model.pth')
|
| 80 |
print("Training complete!")
|
| 81 |
print("Generating response...\n")
|
| 82 |
+
resp = run_model(m, 256, 'JULIET:\nRomeo, Romeo, wherefore art thou Romeo?')
|
| 83 |
+
print("Response:\n" + resp)
|