Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ import transformers
|
|
11 |
from random import randrange
|
12 |
|
13 |
|
14 |
-
|
15 |
|
16 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
17 |
|
@@ -138,5 +138,5 @@ gradio_interface = gradio.Interface(
|
|
138 |
title="FloorAI")
|
139 |
|
140 |
|
141 |
-
gradio_interface.queue(max_size=10, status_update_rate="auto"
|
142 |
gradio_interface.launch(share=True, show_api=True, show_error=True)
|
|
|
11 |
from random import randrange
|
12 |
|
13 |
|
14 |
+
transformers.utils.move_cache()
|
15 |
|
16 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
17 |
|
|
|
138 |
title="FloorAI")
|
139 |
|
140 |
|
141 |
+
gradio_interface.queue(max_size=10, status_update_rate="auto")
|
142 |
gradio_interface.launch(share=True, show_api=True, show_error=True)
|