Update app.py
Browse files
app.py
CHANGED
|
@@ -34,7 +34,7 @@ BEST_STRATEGY_PARAMS = {
|
|
| 34 |
"num_beams": 4,
|
| 35 |
"do_sample": False,
|
| 36 |
"repetition_penalty": 1.2,
|
| 37 |
-
"max_new_tokens":
|
| 38 |
"min_new_tokens": 10,
|
| 39 |
}
|
| 40 |
|
|
@@ -271,7 +271,8 @@ if __name__ == "__main__":
|
|
| 271 |
description="Upload a chest X-ray image to generate a radiology finding in English and automatically translate it to Hindi.",
|
| 272 |
# allow_flagging="never",
|
| 273 |
examples=examples,
|
| 274 |
-
cache_examples=False
|
|
|
|
| 275 |
)
|
| 276 |
|
| 277 |
print("\nStarting Gradio interface...")
|
|
|
|
| 34 |
"num_beams": 4,
|
| 35 |
"do_sample": False,
|
| 36 |
"repetition_penalty": 1.2,
|
| 37 |
+
"max_new_tokens": 100,
|
| 38 |
"min_new_tokens": 10,
|
| 39 |
}
|
| 40 |
|
|
|
|
| 271 |
description="Upload a chest X-ray image to generate a radiology finding in English and automatically translate it to Hindi.",
|
| 272 |
# allow_flagging="never",
|
| 273 |
examples=examples,
|
| 274 |
+
# cache_examples=False
|
| 275 |
+
cache_examples=True
|
| 276 |
)
|
| 277 |
|
| 278 |
print("\nStarting Gradio interface...")
|