Kevin Fink
commited on
Commit
·
3429b72
1
Parent(s):
40b3dde
dev
Browse files
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 |
|