oddadmix commited on
Commit
a171d6f
·
verified ·
1 Parent(s): fd3b370

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,4 +1,4 @@
1
- import spaces
2
 
3
  import gradio as gr
4
  from transformers import pipeline
@@ -21,7 +21,7 @@ except Exception as e:
21
  classifier = None
22
  model_loaded = False
23
 
24
- @spaces.GPU
25
  def classify_prompt(text):
26
  """
27
  Classify a text prompt using the Ara-Prompt-Guard model
 
1
+
2
 
3
  import gradio as gr
4
  from transformers import pipeline
 
21
  classifier = None
22
  model_loaded = False
23
 
24
+
25
  def classify_prompt(text):
26
  """
27
  Classify a text prompt using the Ara-Prompt-Guard model