ashwml commited on
Commit
b626b35
1 Parent(s): b535615

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
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')