bug fixin

app.py

@@ -61,12 +61,15 @@ def self_caption(image):
   generated_ids = model2.generate(pixel_values, max_length=16, num_beams=4, return_dict_in_generate=True)
   # decode into text
   preds = tokenizer.batch_decode(generated_ids[0], skip_special_tokens=True)
+  print("predictions before strip")
+  print(preds)
   preds = [pred.strip() for pred in preds]
   print("Predictions")
   print(preds)
   pred_output = {}
-  for prediction in preds:
-    pred_output[predicted_label] = prediction
+  pred_output = preds
+  #for prediction in preds:
+  #  pred_output[predicted_label] = prediction
   
   return pred_output