Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -55,7 +55,7 @@ def translate(source, source_language, target_language):
|
|
55 |
max_new_tokens = 512-1,
|
56 |
do_sample=False
|
57 |
)
|
58 |
-
predictions = [tokenizer.decode(p, skip_special_tokens=True) for p in predictions.tolist()]
|
59 |
return predictions[0]
|
60 |
|
61 |
|
@@ -101,8 +101,8 @@ with gr.Blocks(theme='sudeepshouche/minimalist') as demo:
|
|
101 |
return {
|
102 |
source: gr.update(interactive=True),
|
103 |
submit: gr.update(interactive=True),
|
104 |
-
source_language: gr.update(interactive=
|
105 |
-
target_language: gr.update(interactive=
|
106 |
}
|
107 |
|
108 |
|
|
|
55 |
max_new_tokens = 512-1,
|
56 |
do_sample=False
|
57 |
)
|
58 |
+
predictions = [tokenizer.decode(p, skip_special_tokens=True).strip() for p in predictions.tolist()]
|
59 |
return predictions[0]
|
60 |
|
61 |
|
|
|
101 |
return {
|
102 |
source: gr.update(interactive=True),
|
103 |
submit: gr.update(interactive=True),
|
104 |
+
source_language: gr.update(interactive=True),
|
105 |
+
target_language: gr.update(interactive=True)
|
106 |
}
|
107 |
|
108 |
|