Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -26,6 +26,7 @@ pipe = pipeline(model = "google/gemma-2-2b-it")
|
|
26 |
def fn(input):
|
27 |
output = pipe(
|
28 |
input,
|
|
|
29 |
)
|
30 |
return output[0]["generated_text"]#[len(input):]
|
31 |
|
|
|
26 |
def fn(input):
|
27 |
output = pipe(
|
28 |
input,
|
29 |
+
max_token = 2048
|
30 |
)
|
31 |
return output[0]["generated_text"]#[len(input):]
|
32 |
|