Update app.py

app.py

@@ -1,5 +1,10 @@
-
-
+import gradio as gr
+import torch
+from torch import nn
+import lightning.pytorch as pl
+from torchvision import datasets
+from torch.nn import functional as F
+from torch.utils.data import DataLoader, Dataset, random_split
 
 def generate_art(character_dropdown, seed_slider):
 
@@ -65,21 +70,21 @@ HTML_TEMPLATE = """
     <!-- Content -->
     <h1>Shakespeare Dialogue Generator</h1>
     <p>Generate new dialogue for Shakespearean character by selecting character from dropdown.</p>
-    <div style="display: flex; justify-content: center; gap: 20px; margin-top: 20px;">
+        <div style="display: flex; justify-content: center; gap: 20px; margin-top: 20px;">
         <div class="concept">
-            <img src="https://github.com/santule/evseq-exp-design-app/assets/20509836/6045f6fc-4d43-4481-9c02-630f34bef465" alt="Romeo">
+            <img src="https://github.com/santule/ERA/assets/20509836/b6b4031a-265d-43f6-bd59-813097c0022b" alt="Romeo">
             <div class="concept-description">Romeo</div>
         </div>
         <div class="concept">
-            <img src="https://github.com/santule/evseq-exp-design-app/assets/20509836/614af361-8e71-43ae-a389-5acb90b06c97" alt="Juliet">
+            <img src="https://github.com/santule/ERA/assets/20509836/4333692c-54f7-42e4-b53a-14044c8151a3" alt="Juliet">
             <div class="concept-description">Juliet</div>
         </div>
         <div class="concept">
-            <img src="https://github.com/santule/evseq-exp-design-app/assets/20509836/bd8abed3-fe27-476f-9aa0-d9b8aa17c1b5" alt="Shakespeare">
+            <img src="https://github.com/santule/ERA/assets/20509836/7e03233a-bf1d-48aa-b156-46f525aa76c6" alt="Shakespeare">
             <div class="concept-description">Shakespeare</div>
         </div>
         <div class="concept">
-            <img src="https://github.com/santule/evseq-exp-design-app/assets/20509836/a1644c6d-4906-417f-b5cc-0114f89e5eea" alt="King Richard III">
+            <img src="https://github.com/santule/ERA/assets/20509836/0dd19c74-72e5-4ea5-8f66-06b7655b81aa" alt="King Richard III">
             <div class="concept-description">King Richard III</div>
         </div>
     </div>