Initial Commit

app.py

@@ -0,0 +1,37 @@
+import gradio as gr
+from huggingface_hub import hf_hub_download
+import torch
+import matplotlib.pyplot as plt
+import numpy as np
+
+path = hf_hub_download('huggan/ArtGAN', 'ArtGAN.pt')
+model = torch.load(path)
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
+
+def generate(seed):
+    with torch.no_grad():
+        noise = torch.randn(seed, 100, 1, 1, device=device)
+    with torch.no_grad():
+        art = model(noise).detach().cpu()
+    gen = np.transpose(art[-1], (1, 2, 0))
+    fig = plt.figure(figsize=(5, 5))
+    plt.imshow(gen)
+    plt.axis('off')
+    return fig
+
+gr.Interface(
+    fn=generate,
+    inputs=[
+        gr.inputs.Slider
+        (
+            label='noise',
+            minimum=10,
+            maximum=100,
+            step=1,
+            default=25
+        )
+    ],  
+    outputs=gr.outputs.Image(type='plot'),
+    title='ArtGAN',
+    description='Generate A Abract Art Using ArtGAN',
+).launch()

requirements.txt

@@ -0,0 +1,5 @@
+gradio==2.6.4
+huggingface_hub==0.5.1
+matplotlib==3.5.0
+numpy==1.21.4
+torch==1.11.0+cu113