ZarahShibli commited on
Commit
de013f3
1 Parent(s): 913a8cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -21,6 +21,7 @@ target_cols = [ 'DevelopmentNotes', 'Expand','Parameters', 'Summary', 'Usage']
21
 
22
 
23
  def predict(comment_sentence):
 
24
  # Set the model in evaluation mode
25
  model.eval()
26
 
@@ -44,7 +45,7 @@ def predict(comment_sentence):
44
 
45
  # Forward pass through the model
46
  with torch.no_grad():
47
- outputs = model(**ids)
48
 
49
 
50
  print(outputs)
 
21
 
22
 
23
  def predict(comment_sentence):
24
+ print(comment_sentence)
25
  # Set the model in evaluation mode
26
  model.eval()
27
 
 
45
 
46
  # Forward pass through the model
47
  with torch.no_grad():
48
+ outputs = model(ids)
49
 
50
 
51
  print(outputs)