Hugging Face's logo

Spaces:
rp-yu
/
apiprompting
Running

App Files Community
yu-rp commited on
Commit
a531efa
·
1 Parent(s): c64fb9f

reduce choices

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -150,7 +150,7 @@ text_highlight_text = gr.Textbox(
150
  type="text")
151
 
152
  radio_api_method = gr.Radio(
153
- ["CLIP_Based API", "LLaVA_Based API"] if 'cuda' in DEVICE else ["CLIP_Based API"],
154
  interactive=True,
155
  value = "CLIP_Based API",
156
  label="Type of API")
 
150
  type="text")
151
 
152
  radio_api_method = gr.Radio(
153
+ ["CLIP_Based API", "LLaVA_Based API"] if torch.cuda.is_available() else ["CLIP_Based API"],
154
  interactive=True,
155
  value = "CLIP_Based API",
156
  label="Type of API")