Hugging Face's logo

Spaces:
vansonhk
/
CLIP-Interrogator
Runtime error

App Files Community
vansonhk commited on
Commit
d30ed46
·
verified ·
1 Parent(s): cf042ff

reverse

Browse files
Files changed (1) hide show
  1. app.py +16 -16
app.py CHANGED
@@ -19,14 +19,14 @@ ci_vitl.clip_model = ci_vitl.clip_model.to("cpu")
19
 
20
  def image_analysis(image, clip_model_name):
21
  # move selected model to GPU and other model to CPU
22
- if clip_model_name == MODELS[0]:
23
- ci_vith.clip_model = ci_vith.clip_model.to("cpu")
24
- ci_vitl.clip_model = ci_vitl.clip_model.to(ci_vitl.device)
25
- ci = ci_vitl
26
- else:
27
- ci_vitl.clip_model = ci_vitl.clip_model.to("cpu")
28
- ci_vith.clip_model = ci_vith.clip_model.to(ci_vith.device)
29
- ci = ci_vith
30
  ci = ci_vitl
31
 
32
  image = image.convert('RGB')
@@ -49,14 +49,14 @@ def image_analysis(image, clip_model_name):
49
 
50
  def image_to_prompt(image, clip_model_name, mode):
51
  # move selected model to GPU and other model to CPU
52
- if clip_model_name == MODELS[0]:
53
- ci_vith.clip_model = ci_vith.clip_model.to("cpu")
54
- ci_vitl.clip_model = ci_vitl.clip_model.to(ci_vitl.device)
55
- ci = ci_vitl
56
- else:
57
- ci_vitl.clip_model = ci_vitl.clip_model.to("cpu")
58
- ci_vith.clip_model = ci_vith.clip_model.to(ci_vith.device)
59
- ci = ci_vith
60
  ci = ci_vitl
61
 
62
  ci.config.blip_num_beams = 64
 
19
 
20
  def image_analysis(image, clip_model_name):
21
  # move selected model to GPU and other model to CPU
22
+ # if clip_model_name == MODELS[0]:
23
+ # ci_vith.clip_model = ci_vith.clip_model.to("cpu")
24
+ # ci_vitl.clip_model = ci_vitl.clip_model.to(ci_vitl.device)
25
+ # ci = ci_vitl
26
+ # else:
27
+ # ci_vitl.clip_model = ci_vitl.clip_model.to("cpu")
28
+ # ci_vith.clip_model = ci_vith.clip_model.to(ci_vith.device)
29
+ # ci = ci_vith
30
  ci = ci_vitl
31
 
32
  image = image.convert('RGB')
 
49
 
50
  def image_to_prompt(image, clip_model_name, mode):
51
  # move selected model to GPU and other model to CPU
52
+ # if clip_model_name == MODELS[0]:
53
+ # ci_vith.clip_model = ci_vith.clip_model.to("cpu")
54
+ # ci_vitl.clip_model = ci_vitl.clip_model.to(ci_vitl.device)
55
+ # ci = ci_vitl
56
+ # else:
57
+ # ci_vitl.clip_model = ci_vitl.clip_model.to("cpu")
58
+ # ci_vith.clip_model = ci_vith.clip_model.to(ci_vith.device)
59
+ # ci = ci_vith
60
  ci = ci_vitl
61
 
62
  ci.config.blip_num_beams = 64