JuanMa360 commited on
Commit
0800af2
·
1 Parent(s): 755057c

refactor: title

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ processor = CLIPProcessor.from_pretrained("tokenizer")
8
 
9
  def generate_answer(image):
10
 
11
- clas = ['high_natural_light','neutral_natural_light','No_Light', 'artificial_lights','others']
12
- classes = ['A picture of a living room filled with abundant natural light with a lot windows, without objects that prevent the light from passing through regardless of whether it is night','A picture of a living room with neutral natural light with few windows, without objects that prevent the light from passing through regardless of whether it is night','Nolights','A picture of a living room with Artificial lights like lamps or headlamps',"Objects or things that are not a room"]
13
  inputs = processor(text=classes, images=image, return_tensors="pt", padding=True)
14
  outputs = model(**inputs)
15
  logits_per_image = outputs.logits_per_image
 
8
 
9
  def generate_answer(image):
10
 
11
+ clas = ['high_natural_light','neutral_natural_light','No_Light', 'artificial_lights']
12
+ classes = ['A picture of a room filled with abundant natural light with a lot windows, without objects that prevent the light from passing through regardless of whether it is night','A picture of a room with neutral natural light with few windows, without objects that prevent the light from passing through regardless of whether it is night','Nolights','A picture of a room with Artificial lights like lamps or headlamps']
13
  inputs = processor(text=classes, images=image, return_tensors="pt", padding=True)
14
  outputs = model(**inputs)
15
  logits_per_image = outputs.logits_per_image