Spaces:
Runtime error
Runtime error
Commit
•
dbb601f
1
Parent(s):
bde2a39
Update app.py
Browse files
app.py
CHANGED
|
@@ -45,7 +45,8 @@ def infer(
|
|
| 45 |
if prompt=="":
|
| 46 |
prompt = " "
|
| 47 |
my_post_dict = {
|
| 48 |
-
"model": "togethercomputer/GPT-JT-6B-v1",
|
|
|
|
| 49 |
"prompt": prompt,
|
| 50 |
"top_p": top_p,
|
| 51 |
"top_k": top_k,
|
|
|
|
| 45 |
if prompt=="":
|
| 46 |
prompt = " "
|
| 47 |
my_post_dict = {
|
| 48 |
+
#"model": "togethercomputer/GPT-JT-6B-v1",
|
| 49 |
+
"model":"EleutherAI/gpt-j-6b",
|
| 50 |
"prompt": prompt,
|
| 51 |
"top_p": top_p,
|
| 52 |
"top_k": top_k,
|