Hugging Face's logo

Spaces:
ktllc
/
Clip-Model
Runtime error

App Files Community
ktllc commited on
Commit
9caa677
·
1 Parent(s): a012f3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -5,8 +5,9 @@ import gradio as gr
5
 
6
  # Load the CLIP model
7
  model, preprocess = clip.load("ViT-B/32")
8
- device = "cuda" if torch.cuda.is available() else "cpu"
9
  model.to(device).eval()
 
10
 
11
  # Define the Business Listing variable
12
  Business_Listing = "Air Guide"
 
5
 
6
  # Load the CLIP model
7
  model, preprocess = clip.load("ViT-B/32")
8
+ device = "cuda" if torch.cuda.is_available() else "cpu"
9
  model.to(device).eval()
10
+ print(device)
11
 
12
  # Define the Business Listing variable
13
  Business_Listing = "Air Guide"