Update app.py

app.py

@@ -47,7 +47,7 @@ def get_screenshot(chat,css=None):
     #css = css
     #hti.screenshot(html_str=html, save_as='red_page.png')
     #hti.screenshot(html_str=html_card, css_str=css, save_as='red_page.png')
-    hti.screenshot(html_str=html, save_as='red_page.png')
+    hti.screenshot(html_str=html, css_str=css, save_as='red_page.png')
     return 'red_page.png',html
 
 
@@ -91,7 +91,7 @@ def chat_inf(system_prompt,prompt,history):
 chat=[('user','bot'),('user','bot')]
 
 #get_screenshot(chat=[('user','bot'),('user','bot')])
-with gr.Blocks() as app:
+with gr.Blocks(css=css) as app:
     with gr.Row():
         with gr.Column(scale=3):
             with gr.Group():