asoria HF staff commited on
Commit
1628a30
1 Parent(s): b62bc45

try to fix GPU error

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -1,3 +1,7 @@
 
 
 
 
1
  import logging
2
  import os
3
 
@@ -24,11 +28,6 @@ from transformers import (
24
  pipeline,
25
  )
26
  from torch import bfloat16
27
-
28
- # These imports at the end because of torch/datamapplot issue in Zero GPU
29
- import spaces
30
- import gradio as gr
31
-
32
  from prompts import REPRESENTATION_PROMPT
33
 
34
  """
 
1
+ # These imports at the end because of torch/datamapplot issue in Zero GPU
2
+ import spaces
3
+ import gradio as gr
4
+
5
  import logging
6
  import os
7
 
 
28
  pipeline,
29
  )
30
  from torch import bfloat16
 
 
 
 
 
31
  from prompts import REPRESENTATION_PROMPT
32
 
33
  """