Kevin Fink commited on
Commit
3429b72
·
1 Parent(s): 40b3dde
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -254,6 +254,8 @@ def predict(text):
254
  with torch.no_grad(): # Disable gradient calculation for inference
255
  outputs = model.generate(inputs)
256
  predictions = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
 
257
  return predictions
258
 
259
 
 
254
  with torch.no_grad(): # Disable gradient calculation for inference
255
  outputs = model.generate(inputs)
256
  predictions = tokenizer.decode(outputs[0], skip_special_tokens=True)
257
+ print('xxxxxxxxxxxxxxxxxxxxxxx')
258
+ print(predictions)
259
  return predictions
260
 
261