Update app.py
Browse files
app.py
CHANGED
@@ -99,9 +99,7 @@ def predict_event(image):
|
|
99 |
def update_metrics():
|
100 |
image="cutegirl.jpeg"
|
101 |
generated_caption = tokenizer.decode(model.generate(feature_extractor(Image.open(image), return_tensors="pt",max_length=16,num_beams = 4).pixel_values.to(device))[0])
|
102 |
-
|
103 |
-
dict_metric_scores = {}
|
104 |
-
dict_metrics = {"rouge2": evaluate.load("rouge"),
|
105 |
"bleu": evaluate.load('bleu'),
|
106 |
"bertscore": evaluate.load("bertscore"),
|
107 |
"meteor": evaluate.load('meteor')
|
|
|
99 |
def update_metrics():
|
100 |
image="cutegirl.jpeg"
|
101 |
generated_caption = tokenizer.decode(model.generate(feature_extractor(Image.open(image), return_tensors="pt",max_length=16,num_beams = 4).pixel_values.to(device))[0])
|
102 |
+
dict_metrics = {"rouge2": evaluate.load("rouge"),
|
|
|
|
|
103 |
"bleu": evaluate.load('bleu'),
|
104 |
"bertscore": evaluate.load("bertscore"),
|
105 |
"meteor": evaluate.load('meteor')
|