Kevin Fink
commited on
Commit
·
8a66ad2
1
Parent(s):
b707c9c
dev
Browse files
app.py
CHANGED
|
@@ -267,6 +267,7 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
|
|
| 267 |
return f"An error occurred: {str(e)}, TB: {traceback.format_exc()}"
|
| 268 |
return 'DONE!'#train_result
|
| 269 |
|
|
|
|
| 270 |
def test(text):
|
| 271 |
from transformers import pipeline
|
| 272 |
model_name = 'shorecode/t5-efficient-tiny-nh8-summarizer'
|
|
|
|
| 267 |
return f"An error occurred: {str(e)}, TB: {traceback.format_exc()}"
|
| 268 |
return 'DONE!'#train_result
|
| 269 |
|
| 270 |
+
@spaces.GPU
|
| 271 |
def test(text):
|
| 272 |
from transformers import pipeline
|
| 273 |
model_name = 'shorecode/t5-efficient-tiny-nh8-summarizer'
|