Update app.py

app.py

@@ -39,14 +39,11 @@ class Solution(object):
 
 textbox2 = gr.Textbox()
 
-image_code = gr.Image(value = "https://huggingface.co/spaces/PinoCorgi/CodeExplainerPython/blob/main/code.jpg")
-image_output = gr.Image(value = "https://huggingface.co/spaces/PinoCorgi/CodeExplainerPython/blob/main/output.jpg")
-
-demo = gr.Interface(fn = generate_text, inputs = textbox1, outputs = textbox2)
-
-image_code
-image_output
-
 if __name__ == "__main__":
-    print(f"The Inference will take approximately 1 min 30 Seconds. Attached are example images of input and output.")
+    with gr.Blocks() as demo:
+        gr.TextBox("The Inference Takes about 1 min 30 seconds")
+        gr.Interface(fn = generate_text, inputs = textbox1, outputs = textbox2)
+        with gr.Row():
+            gr.Image(value = "code.jpg")
+            gr.Image(value = "output.jpg")
     demo.launch()