Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -21,12 +21,12 @@ app = gr.load(
|
|
21 |
import gradio as gr
|
22 |
from transformers import pipeline
|
23 |
|
24 |
-
pipe = pipeline(model = "google/gemma-
|
25 |
|
26 |
def fn(input):
|
27 |
output = pipe(
|
28 |
input,
|
29 |
-
max_new_tokens=2048
|
30 |
)
|
31 |
return output[0]["generated_text"]#[len(input):]
|
32 |
|
|
|
21 |
import gradio as gr
|
22 |
from transformers import pipeline
|
23 |
|
24 |
+
pipe = pipeline(model = "google/gemma-1.1-7b-it")
|
25 |
|
26 |
def fn(input):
|
27 |
output = pipe(
|
28 |
input,
|
29 |
+
max_new_tokens = 2048
|
30 |
)
|
31 |
return output[0]["generated_text"]#[len(input):]
|
32 |
|