Update app.py

app.py

@@ -203,7 +203,7 @@ def replace_nsfw_images(results):
 
 with gr.Blocks(css="style.css") as demo:
     gr.HTML(
-        """
+        f"""
             <div class="finetuned-diffusion-div">
               <div>
                 <h1>Protogen Diffusion</h1>
@@ -211,6 +211,9 @@ with gr.Blocks(css="style.css") as demo:
               <p>
                Demo for multiple fine-tuned Protogen Stable Diffusion models.
               </p>
+              <p>
+               Running on <b>{device}</b>
+              </p>
               <p>You can also duplicate this space and upgrade to gpu by going to settings:<br>
               <a style="display:inline-block" href="https://huggingface.co/spaces/patrickvonplaten/finetuned_diffusion?duplicate=true"><img src="https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14" alt="Duplicate Space"></a></p>
             </div>