geolocation87 commited on
Commit
717c137
·
verified ·
1 Parent(s): 1272f45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import torch
 
2
  if torch.cuda.is_available():
3
  model = GeoCLIP().to("cuda")
4
  else:
 
1
  import torch
2
+ from geoclip import GeoCLIP
3
  if torch.cuda.is_available():
4
  model = GeoCLIP().to("cuda")
5
  else: