Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,11 +1,9 @@
|
|
| 1 |
from transformers import pipeline
|
| 2 |
-
import torch # Import torch library
|
| 3 |
|
| 4 |
# Use the GPT-Neo model, a larger and more capable alternative to GPT-2
|
| 5 |
model = pipeline(
|
| 6 |
"text-generation",
|
| 7 |
model="EleutherAI/gpt-neo-1.3B", # GPT-Neo 1.3B model
|
| 8 |
-
torch_dtype=torch.float16 # Reduce memory usage
|
| 9 |
)
|
| 10 |
|
| 11 |
def generate_test_cases(requirement):
|
|
|
|
| 1 |
from transformers import pipeline
|
|
|
|
| 2 |
|
| 3 |
# Use the GPT-Neo model, a larger and more capable alternative to GPT-2
|
| 4 |
model = pipeline(
|
| 5 |
"text-generation",
|
| 6 |
model="EleutherAI/gpt-neo-1.3B", # GPT-Neo 1.3B model
|
|
|
|
| 7 |
)
|
| 8 |
|
| 9 |
def generate_test_cases(requirement):
|