biplab2008 commited on
Commit
4b08332
·
verified ·
1 Parent(s): 75280f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -358,7 +358,10 @@ def pil_parser(video_file):
358
  X.append(image)
359
  X = torch.stack(X, dim=1).unsqueeze(0)
360
 
361
- return 'shape is : '+ str(X.shape)
 
 
 
362
 
363
  example_list=[
364
  ["2350.mp4"],
 
358
  X.append(image)
359
  X = torch.stack(X, dim=1).unsqueeze(0)
360
 
361
+ out = model(X)
362
+
363
+ #return 'shape is : '+ str(X.shape)
364
+ return 'viscosity : ' + str(round(out.item(),1)) + ' cp'
365
 
366
  example_list=[
367
  ["2350.mp4"],