temp-9384289 commited on
Commit
647ce28
1 Parent(s): 20c2904

models statement

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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