Update app.py

app.py

@@ -1,5 +1,6 @@
 import keras
 import gradio as gr
+from translate import Translator
 from huggingface_hub import hf_hub_download
 
 link = hf_hub_download(repo_id="DiDiR6/GPT2Financial", filename="GPT2Financial_model.keras")
@@ -8,10 +9,25 @@ gpt2 = keras.models.load_model(link)
 gpt2.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])
 
 def respond(message, chat_history):
+    translator = Translator(to_lang="en")
+    translation = translator.translate(message)
     bot_message = gpt2.generate(message)
     return bot_message
 
 
+with gr.Blocks() as app:
+    gr.HTML("""
+        <div style='width: 100%; height: 300px; background: url("https://raw.githubusercontent.com/sxfiene/Finance_GPT/main/logo.png") no-repeat center center;'>
+        </div>
+        <h1 style='text-align:center; width=100%; margin-top=500px'>FinanceGPT</h1>
+        <h3 style='text-align:center;'>L'IA répondant à vos intérogations sur la finance</h3>
+    """)
+    with gr.Row():
+        question = gr.Textbox(label="Qu'est ce que voulez-vous savoir ?", placeholder="Saisissez ici votre demande...")
+        submit = gr.Button("Envoyer")
+    response = gr.Textbox(label="Réponse", interactive=False)
+    
 
-demo = gr.ChatInterface(fn=respond, title="GPT2 Financial", retry_btn=None, undo_btn=None, clear_btn=None)
-demo.launch(height=650)
+    submit.click(fn=respond)
+
+app.launch()