Spaces:
Sleeping
Sleeping
gt
Browse files
app.py
CHANGED
|
@@ -36,7 +36,7 @@ def get_completion(prompt):
|
|
| 36 |
|
| 37 |
def greet(input):
|
| 38 |
response = get_completion(input)
|
| 39 |
-
gen_text = response[0]
|
| 40 |
return gen_text
|
| 41 |
#return json.dumps(response)
|
| 42 |
|
|
|
|
| 36 |
|
| 37 |
def greet(input):
|
| 38 |
response = get_completion(input)
|
| 39 |
+
gen_text = response["predictions"][0]["generated_text"]
|
| 40 |
return gen_text
|
| 41 |
#return json.dumps(response)
|
| 42 |
|