devic1 commited on
Commit
6061089
·
1 Parent(s): 84da857

added float numbers as result

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -37,9 +37,7 @@ def weightsbias(weight,bias,predicted,progress=gr.Progress()):
37
  t = torch.tensor([predicted],dtype=torch.float32)
38
  rout = model(t)
39
  c = model.state_dict()
40
- def ceil(x):
41
- return math.ceil(x)
42
- return [ceil(rout.item()),ceil(c['l.weight'].item()),ceil(c['l.bias'].item())]
43
 
44
 
45
  with gr.Blocks() as iface:
 
37
  t = torch.tensor([predicted],dtype=torch.float32)
38
  rout = model(t)
39
  c = model.state_dict()
40
+ return [rout.item(),c['l.weight'].item(),c['l.bias'].item()]
 
 
41
 
42
 
43
  with gr.Blocks() as iface: