jhoppanne commited on
Commit
2732d8a
1 Parent(s): 437d7d5

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. .ipynb_checkpoints/main-checkpoint.py +1 -1
  2. main.py +1 -1
.ipynb_checkpoints/main-checkpoint.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- pipeline = pipeline("image-classification", model="jhoppanne/SkinCancerClassifier_Plain-V0")
5
 
6
  def predict(input_img):
7
  pred = pipeline(input_img)
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ pipeline = pipeline("image-classification", model="jhoppanne/SkinCancerClassifier_smote-V0")
5
 
6
  def predict(input_img):
7
  pred = pipeline(input_img)
main.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- pipeline = pipeline("image-classification", model="jhoppanne/SkinCancerClassifier_Plain-V0")
5
 
6
  def predict(input_img):
7
  pred = pipeline(input_img)
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ pipeline = pipeline("image-classification", model="jhoppanne/SkinCancerClassifier_smote-V0")
5
 
6
  def predict(input_img):
7
  pred = pipeline(input_img)