Update README.md
Browse files
README.md
CHANGED
|
@@ -93,6 +93,7 @@ prompt = f"Generate unit tests in Dart for the following class:\n{input_code}"
|
|
| 93 |
inputs = tokenizer(prompt, return_tensors="pt")
|
| 94 |
outputs = model.generate(**inputs, max_length=512)
|
| 95 |
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
|
|
|
| 96 |
|
| 97 |
## Training Details
|
| 98 |
|
|
|
|
| 93 |
inputs = tokenizer(prompt, return_tensors="pt")
|
| 94 |
outputs = model.generate(**inputs, max_length=512)
|
| 95 |
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
| 96 |
+
```
|
| 97 |
|
| 98 |
## Training Details
|
| 99 |
|