Spaces:
Running
Running
temp-9384289
commited on
Commit
•
647ce28
1
Parent(s):
20c2904
models statement
Browse files
app.py
CHANGED
@@ -21,7 +21,7 @@ import base64
|
|
21 |
modelieo=[
|
22 |
'nathanReitinger/MNIST-diffusion',
|
23 |
'nathanReitinger/MNIST-diffusion-oneImage',
|
24 |
-
'nathanReitinger/MNIST-diffusion-threeHundredImages
|
25 |
'nathanReitinger/MNIST-GAN',
|
26 |
'nathanReitinger/MNIST-GAN-noDropout',
|
27 |
'nathanReitinger/FASHION-diffusion',
|
@@ -251,10 +251,10 @@ def TextToImage(Prompt,inference_steps, model):
|
|
251 |
return [ai_gen, another_one]
|
252 |
|
253 |
df = pd.DataFrame({
|
254 |
-
"Model" : ['MNIST-diffusion', 'MNIST-diffusion-oneImage', 'MNIST-GAN', 'MNIST-GAN-noDropout', 'FASHION-diffuion-oneImage', 'FASHION-diffusion'],
|
255 |
-
"Class (Architecture)" : ['UNet2DModel', 'UNet2DModel', 'Sequential', 'Sequential', 'UNet2DModel', 'UNet2DModel'],
|
256 |
-
"Dataset Examples" : [60000, 1, 60000, 60000, 1, 60000],
|
257 |
-
"Notes" : ['Similar architecture as Stable Diffusion, different training data', 'Toy model, purposed to store protected content', 'GANs are not as likely to store protected content', 'less dropout, more copying?', 'same diffusion, different data (more variance in data)','larger diffusion training data, on FASHION dataset']
|
258 |
})
|
259 |
|
260 |
# Applying style to highlight the maximum value in each row
|
|
|
21 |
modelieo=[
|
22 |
'nathanReitinger/MNIST-diffusion',
|
23 |
'nathanReitinger/MNIST-diffusion-oneImage',
|
24 |
+
'nathanReitinger/MNIST-diffusion-threeHundredImages',
|
25 |
'nathanReitinger/MNIST-GAN',
|
26 |
'nathanReitinger/MNIST-GAN-noDropout',
|
27 |
'nathanReitinger/FASHION-diffusion',
|
|
|
251 |
return [ai_gen, another_one]
|
252 |
|
253 |
df = pd.DataFrame({
|
254 |
+
"Model" : ['MNIST-diffusion', 'MNIST-diffusion-oneImage', 'MNIST-diffusion-threeHundredImages', 'MNIST-GAN', 'MNIST-GAN-noDropout', 'FASHION-diffuion-oneImage', 'FASHION-diffusion'],
|
255 |
+
"Class (Architecture)" : ['UNet2DModel', 'UNet2DModel', 'UNet2DModel', 'Sequential', 'Sequential', 'UNet2DModel', 'UNet2DModel'],
|
256 |
+
"Dataset Examples" : [60000, 1, 300, 60000, 60000, 1, 60000],
|
257 |
+
"Notes" : ['Similar architecture as Stable Diffusion, different training data', 'Toy model, purposed to store protected content', 'testing Toy model with slightly larger dataset', 'GANs are not as likely to store protected content', 'less dropout, more copying?', 'same diffusion, different data (more variance in data)','larger diffusion training data, on FASHION dataset']
|
258 |
})
|
259 |
|
260 |
# Applying style to highlight the maximum value in each row
|