Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from huggingface_hub import hf_hub_download
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
@@ -28,8 +29,12 @@ class TinySR(nn.Module):
         return self.output(x)
 
 # --- 2️⃣ Load trained model ---
+weights_path = hf_hub_download(
+    repo_id="MihaiPopa-1/TinySR",
+    filename="tinysr_model.pt"
+)
 model = TinySR(channels=32, blocks=8)
-model.load_state_dict(torch.load("tinysr_model.pt", map_location="cpu"))
+model.load_state_dict(torch.load(weights_path, map_location="cpu"))
 model.eval()
 
 # --- 3️⃣ Inference function ---
@@ -62,8 +67,7 @@ iface = gr.Interface(
     inputs=gr.Audio(type="filepath", label="Upload 16 kHz Mono WAV"),
     outputs=gr.Audio(type="filepath", label="Upscaled 48 kHz WAV"),
     title=title,
-    description=description,
-    allow_flagging="never"
+    description=description
 )
 
 iface.launch()