change response and reformat code

app.py

@@ -1,5 +1,6 @@
 from image import *
 import streamlit as st
+
 # url = "https://i.imgur.com/qs0CxjE_d.webp?maxwidth=760&fidelity=grand"
 # text = "What is the two color's car ?"
 
@@ -8,5 +9,5 @@ st.write('Application starting...')
 text = st.text_input('Posez votre question (en anglais)')
 url = st.text_input('mettez le liens de votre image')
 
-if st.button('générer') :
-    st.write(image(url, text))
+if st.button('générer'):
+    st.write('response is :', image(url, text))

image.py

@@ -1,6 +1,8 @@
 from transformers import ViltProcessor, ViltForQuestionAnswering
 import requests
 from PIL import Image
+
+
 def image(url, text):
     image = Image.open(requests.get(url, stream=True).raw)
     processor = ViltProcessor.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
@@ -17,9 +19,6 @@ def image(url, text):
     print("image link:", url)
     print("Predicted answer:", model.config.id2label[idx])
 
-    return "Predicted answer:", model.config.id2label[idx]
-
-
+    return model.config.id2label[idx]
 
 # prepare image + question
-