ARCQUB commited on
Commit
15a946c
·
verified ·
1 Parent(s): 72c589b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -2,6 +2,9 @@ import gradio as gr
2
  import importlib
3
  from PIL import Image
4
  import json
 
 
 
5
 
6
  # === Model Mapping ===
7
  MODEL_MAP = {
@@ -173,4 +176,7 @@ iface = gr.Interface(
173
  allow_flagging="never"
174
  )
175
 
176
- iface.launch(share=True)
 
 
 
 
2
  import importlib
3
  from PIL import Image
4
  import json
5
+ import spaces
6
+
7
+ @spaces.GPU
8
 
9
  # === Model Mapping ===
10
  MODEL_MAP = {
 
176
  allow_flagging="never"
177
  )
178
 
179
+ if __name__ == "__main__":
180
+ main()
181
+
182
+ #iface.launch(share=True)