bug fixing

app.py

@@ -59,19 +59,20 @@ def self_caption(image):
   print(pixel_values)
   # autoregressively generate text (using beam search or other decoding strategy)
   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)
+  #generated_sentences = tokenizer.batch_decode(encoder_outputs, skip_special_tokens=True)
+
+  #return(generated_sentences[0].split('.')[0])
+
   preds = [pred.strip() for pred in preds]
   print("Predictions")
   print(preds)
-  pred_output = {}
-  pred_output = preds
-  #for prediction in preds:
-  #  pred_output[predicted_label] = prediction
+
   
-  return preds[0]
+  return(preds[0].split('.')[0])
 
 def classify_image(image):
   results = image_pipe(image)
@@ -108,7 +109,10 @@ print("img_info2")
 img_info2 = gr.Interface(
     fn=self_caption,
     inputs=image,
-    outputs=label,
+    #outputs=label,
+    outputs = [
+    gr.outputs.Textbox(label = 'Caption')
+],
 )
 print("running parallel call")
 Parallel(