JuanMa360 commited on
Commit
3c0126a
·
1 Parent(s): 0800af2

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']
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
 
8
 
9
  def generate_answer(image):
10
 
11
+ classes = ['A picture of a room filled with abundant natural light with a lot or few windows regardless of whether it is night, without objects that prevent the light from passing through.','a picture of room in the dark','A picture of a living room with Artificial lights like lamps or headlamps']
12
+ clas = ['natural_light','no_light', 'artificial_light']
13
  inputs = processor(text=classes, images=image, return_tensors="pt", padding=True)
14
  outputs = model(**inputs)
15
  logits_per_image = outputs.logits_per_image