Omnibus commited on
Commit
8a36771
·
verified ·
1 Parent(s): 14a7ff3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -15,11 +15,12 @@ def list_files(model_name):
15
 
16
  def load_bin(model_name,file_name):
17
  r = requests.get(f'https://huggingface.co/{model_name}/resolve/main/{file_name}')
18
- print(r.json)
19
  print("#######")
20
  print(dir(r))
21
  #result=torch.frombuffer(r,dtype=torch.get_default_dtype())
22
- result = torch.frombuffer(io.BytesIO(r.json),dtype=torch.get_default_dtype())
 
23
  return result
24
  #with open()
25
  #file = open(f'{name}/{f_name}','wb')
 
15
 
16
  def load_bin(model_name,file_name):
17
  r = requests.get(f'https://huggingface.co/{model_name}/resolve/main/{file_name}')
18
+ print(r.content)
19
  print("#######")
20
  print(dir(r))
21
  #result=torch.frombuffer(r,dtype=torch.get_default_dtype())
22
+ #result = torch.frombuffer(io.BytesIO(),dtype=torch.get_default_dtype())
23
+ result = torch.from_numpy(r.content)
24
  return result
25
  #with open()
26
  #file = open(f'{name}/{f_name}','wb')