Spaces:
Runtime error
Runtime error
Commit
•
ede6cee
1
Parent(s):
33669c4
Update app.py
Browse files
app.py
CHANGED
@@ -57,7 +57,7 @@ def query(payload):
|
|
57 |
return response.json()
|
58 |
|
59 |
|
60 |
-
def inference(input_sentence, max_length, sample_or_greedy, raw_text=
|
61 |
payload = {
|
62 |
"prompt": input_sentence,
|
63 |
"do_sample": True,
|
@@ -68,7 +68,8 @@ def inference(input_sentence, max_length, sample_or_greedy, raw_text=False):
|
|
68 |
payload
|
69 |
)
|
70 |
|
71 |
-
if raw_text:
|
|
|
72 |
return None, data[0]['generated_text']
|
73 |
|
74 |
width, height = 3326, 3326
|
|
|
57 |
return response.json()
|
58 |
|
59 |
|
60 |
+
def inference(input_sentence, max_length, sample_or_greedy, raw_text=True):
|
61 |
payload = {
|
62 |
"prompt": input_sentence,
|
63 |
"do_sample": True,
|
|
|
68 |
payload
|
69 |
)
|
70 |
|
71 |
+
# if raw_text:
|
72 |
+
if True:
|
73 |
return None, data[0]['generated_text']
|
74 |
|
75 |
width, height = 3326, 3326
|