Update app.py

Corrected a stupid mistake

app.py

@@ -26,14 +26,11 @@ def load_dit_model(dit_size):
     
     # Configure model based on size
     if dit_size == "S":
-        model = DiT(num_blocks=8, hidden_size=384, num_heads=6)
-        model.from_pretrained("kaupane/DiT-Wikiart-Small")
+        model = DiT.from_pretrained("kaupane/DiT-Wikiart-Small")
     elif dit_size == "B":
-        model = DiT(num_blocks=12, hidden_size=640, num_heads=10)
-        model.from_pretrained("kaupane/DiT-Wikiart-Base")
+        model = DiT.from_pretrained("kaupane/DiT-Wikiart-Base")
     elif dit_size == "L":
-        model = DiT(num_blocks=16, hidden_size=896, num_heads=14)
-        model.from_pretrained("kaupane/DiT-Wikiart-Large")
+        model = DiT.from_pretrained("kaupane/DiT-Wikiart-Large")
     else:
         raise ValueError(f"Invalid DiT size: {dit_size}")