Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import numpy
 import torch
 import requests
+import io
 from huggingface_hub import HfApi, ModelFilter, list_models, list_repo_files, hf_hub_download
 api = HfApi()
 
@@ -14,8 +15,8 @@ def list_files(model_name):
 
 def load_bin(model_name,file_name):
     r = requests.get(f'https://huggingface.co/{model_name}/resolve/main/{file_name}')
-    result=torch.frombuffer(r,dtype=torch.get_default_dtype())
-
+    #result=torch.frombuffer(r,dtype=torch.get_default_dtype())
+    result = torch.load(io.BytesIO(r.content))
     return result
     #with open()
     #file = open(f'{name}/{f_name}','wb')