Update app.py

app.py

@@ -93,12 +93,13 @@ def infer(prompt,
         "early_stopping":early_stopping,
         "no_repeat_ngram_size":no_repeat_ngram_size,
         "num_beams":num_beams,
-        "return_full_text":return_full_text
+        "return_full_text":return_full_text,
+        "raw_response":True
     }
     
     s = time.time()
     response = inference(prompt, params=params)
-    #print(response)
+    print(response)
     proc_time = time.time()-s
     #print(f"Processing time was {proc_time} seconds")
     return response
@@ -166,11 +167,11 @@ with gr.Blocks() as demo:
         """Enter a prompt  and get the tee shirt design. Use examples as a guide. We use an equally powerful AI model bigscience/bloom."""
         )
     with gr.Row() as row:
-      with gr.Column():
-        textbox = gr.Textbox(placeholder="Enter prompt...", lines=1,label='Your prompt')
-        btn = gr.Button("Generate")    
-    with gr.Column():
-        output_image = gr.components.Image(label="Your tee shirt")
+        with gr.Column():
+            textbox = gr.Textbox(placeholder="Enter prompt...", lines=1,label='Your prompt')
+            btn = gr.Button("Generate")    
+        with gr.Column():
+            output_image = gr.components.Image(label="Your tee shirt")
 
 
     btn.click(getadvertisement,inputs=[textbox], outputs=[output_image])