Hugging Face's logo

Spaces:
kevinhug
/
clientX
Sleeping

App Files Community
kevinhug commited on
Commit
b176344
·
1 Parent(s): b27e17f

fine tune LLM

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -28,8 +28,8 @@ def similar(issue):
28
  '''
29
  FINE TUNE LLM LIKE SCORE
30
  '''
31
- from fastai.vision.all import *
32
- #learn = load_learner('banks_txt_like.pkl')
33
  def like(issue):
34
  pred,idx,probs = learn.predict(issue)
35
  return pred
 
28
  '''
29
  FINE TUNE LLM LIKE SCORE
30
  '''
31
+ from fastai.text.all import *
32
+ learn = load_learner('./banks_txt_like.pkl')
33
  def like(issue):
34
  pred,idx,probs = learn.predict(issue)
35
  return pred