update app and image.

app.py

@@ -1,13 +1,18 @@
 from image import *
 import streamlit as st
+import os
 
 # url = "https://i.imgur.com/qs0CxjE_d.webp?maxwidth=760&fidelity=grand"
 # text = "What is the two color's car ?"
 
-st.write('Application starting...')
+st.write('Part 1')
 
 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('response is :', image(url, text))
+
+st.write('Part 2')
+
+os.system("language.py")

image.py

@@ -15,9 +15,9 @@ def image(url, text):
     outputs = model(**encoding)
     logits = outputs.logits
     idx = logits.argmax(-1).item()
-    print("question asked:", text)
-    print("image link:", url)
-    print("Predicted answer:", model.config.id2label[idx])
+    # print("question asked:", text)
+    # print("image link:", url)
+    # print("Predicted answer:", model.config.id2label[idx])
 
     return model.config.id2label[idx]
 

language.py

@@ -0,0 +1,9 @@
+from transformers import pipeline
+
+input = "i have question and answere.\
+the question is : How many cars here ?\
+the response is : 2\
+with this information, can you crate an answere phrase in french"
+
+generator = pipeline('text-generation', model='gpt2')
+print(generator(input, max_length=30))