CesarLeblanc commited on
Commit
ee21ab1
1 Parent(s): edd2cf0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -84,8 +84,8 @@ def classification(text, typology, confidence):
84
  def masking(text):
85
  text = gbif_normalization(text)
86
  masked_text = text + ', [MASK]'
87
- pred = mask_model(masked_text)[0]
88
- text = f"The last species from this vegetation plot is probably {pred}."
89
  image = return_species_image(new_species)
90
  return text, image
91
 
 
84
  def masking(text):
85
  text = gbif_normalization(text)
86
  masked_text = text + ', [MASK]'
87
+ new_species = mask_model(masked_text)[0]
88
+ text = f"The last species from this vegetation plot is probably {new_species}."
89
  image = return_species_image(new_species)
90
  return text, image
91