CodeCommunity commited on
Commit
a6e09c7
·
verified ·
1 Parent(s): 2781db5

Update app/predictor.py

Browse files
Files changed (1) hide show
  1. 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()