Spaces:
Running
Running
Update app/predictor.py
Browse files- app/predictor.py +3 -1
app/predictor.py
CHANGED
|
@@ -4,6 +4,7 @@ import logging
|
|
| 4 |
import torch
|
| 5 |
import torch.nn.functional as F
|
| 6 |
from transformers import AutoModel, AutoTokenizer
|
|
|
|
| 7 |
|
| 8 |
# just use to log at terminal no big deal here
|
| 9 |
logging.basicConfig(level=logging.INFO)
|
|
@@ -736,4 +737,5 @@ class GuideGenerator:
|
|
| 736 |
|
| 737 |
# Create the global instance
|
| 738 |
classifier = CodeClassifier()
|
| 739 |
-
guide_generator = GuideGenerator()
|
|
|
|
|
|
| 4 |
import torch
|
| 5 |
import torch.nn.functional as F
|
| 6 |
from transformers import AutoModel, AutoTokenizer
|
| 7 |
+
from app.services.reviewer_service import AIReviewerService
|
| 8 |
|
| 9 |
# just use to log at terminal no big deal here
|
| 10 |
logging.basicConfig(level=logging.INFO)
|
|
|
|
| 737 |
|
| 738 |
# Create the global instance
|
| 739 |
classifier = CodeClassifier()
|
| 740 |
+
guide_generator = GuideGenerator()
|
| 741 |
+
reviewer = AIReviewerService()
|