LuyangZ commited on
Commit
c1197f5
·
verified ·
1 Parent(s): 0326c77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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", api_open=True)
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)