Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -96,9 +96,9 @@ def categorize(input):
|
|
96 |
return ttc(input)[0]["generated_text"]
|
97 |
|
98 |
@spaces.GPU
|
99 |
-
def turna(input, max_new_tokens
|
100 |
-
top_k
|
101 |
-
do_sample
|
102 |
turna = pipeline(model="boun-tabi-LMG/TURNA", device=0)
|
103 |
|
104 |
return turna(input, max_new_tokens = max_new_tokens, length_penalty=length_penalty,
|
|
|
96 |
return ttc(input)[0]["generated_text"]
|
97 |
|
98 |
@spaces.GPU
|
99 |
+
def turna(input, max_new_tokens, length_penalty,
|
100 |
+
top_k, top_p, temp, num_beams,
|
101 |
+
do_sample):
|
102 |
turna = pipeline(model="boun-tabi-LMG/TURNA", device=0)
|
103 |
|
104 |
return turna(input, max_new_tokens = max_new_tokens, length_penalty=length_penalty,
|