Update app.py

app.py

@@ -1,11 +1,9 @@
 import gradio as gr
 import torch
 from diffusers.models import UNet2DModel
-from huggingface_hub import hf_hub_url, cached_download
+from huggingface_hub import hf_hub_download
 
-
-config_file_url = hf_hub_url(repo_id="porestar/oadg_channels_64", filename="model.pt", revision="main")
-cached_download(config_file_url)
+path = hf_hub_download(repo_id="porestar/oadg_channels_64", filename="model.pt")
 
 model = UNet2DModel(
     sample_size=64,
@@ -27,7 +25,7 @@ model = UNet2DModel(
     ),
 )
 
-model.load_state_dict(torch.load("./oadg_channels_64/model.pt"))
+model.load_state_dict(torch.load(path))
 
 
 def classify_image(inp):