SengTak commited on
Commit
0a98cc1
·
1 Parent(s): 5062089

Update space fixed l3prune import

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,6 +1,9 @@
1
  import gradio as gr
2
  import joblib
3
  import torch
 
 
 
4
  from l3prune import LLMEncoder
5
  #load the model
6
  best_clf = joblib.load("./saved/classifier_llama32.joblib")
 
1
  import gradio as gr
2
  import joblib
3
  import torch
4
+ import sys
5
+ import os
6
+ sys.path.append(os.path.dirname(os.path.abspath(__file__)))
7
  from l3prune import LLMEncoder
8
  #load the model
9
  best_clf = joblib.load("./saved/classifier_llama32.joblib")