Update app.py

app.py

@@ -14,7 +14,7 @@ random_wt = random.choice(wts)
 
 # Load trained model weights from Hugging Face Hub
 weights_path = hf_hub_download(
-    repo_id="keysun89/img_generation",  # Fixed repo name
+    repo_id="keysun89/image_generation",  # Fixed repo name
     filename=random_wt
 )
 
@@ -40,7 +40,7 @@ model.eval()
 
 wt_2 = 'genrator.pth'
 srgan_weights = hf_hub_download(
-    repo_id="keysun89/img_generation",  # Fixed repo name
+    repo_id="keysun89/image_generation",  # Fixed repo name
     filename=wt_2
 )