ginod22 commited on
Commit
9d5882a
·
1 Parent(s): d7e902b

Revert "fixing bugs?"

Browse files

This reverts commit d7e902b0fc3c3324931fa917b076c37b5b5a4efb.

Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +1 -1
app.py CHANGED
@@ -1,11 +1,11 @@
1
  import gradio as gr
2
- import joblib
3
  import numpy as np
4
  from PIL import Image
5
 
6
  # Load your model
7
  with open("model.pkl", "rb") as f:
8
- model = joblib.load(f)
9
 
10
  # Preprocessing function
11
  def preprocess_image(img: Image.Image):
 
1
  import gradio as gr
2
+ import pickle
3
  import numpy as np
4
  from PIL import Image
5
 
6
  # Load your model
7
  with open("model.pkl", "rb") as f:
8
+ model = pickle.load(f)
9
 
10
  # Preprocessing function
11
  def preprocess_image(img: Image.Image):
requirements.txt CHANGED
@@ -2,4 +2,4 @@ gradio==4.26.0
2
  scikit-learn==1.4.2
3
  pillow==10.3.0
4
  numpy==1.26.4
5
- joblib==1.4.2
 
2
  scikit-learn==1.4.2
3
  pillow==10.3.0
4
  numpy==1.26.4
5
+