ErnestBeckham commited on
Commit
00c17d4
·
1 Parent(s): 8635d04

app updated

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -2,7 +2,6 @@ import streamlit as st
2
  import tensorflow as tf
3
  import cv2
4
  import numpy as np
5
- from patchify import patchify
6
  from huggingface_hub import from_pretrained_keras
7
 
8
  m = from_pretrained_keras('ErnestBeckham/ViT-Lungs')
@@ -11,7 +10,7 @@ hp = {}
11
  hp['class_names'] = ["lung_aca", "lung_n", "lung_scc"]
12
 
13
  def main():
14
- st.title("Multi-Cancer Classification")
15
 
16
  # Upload image through drag and drop
17
  uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
 
2
  import tensorflow as tf
3
  import cv2
4
  import numpy as np
 
5
  from huggingface_hub import from_pretrained_keras
6
 
7
  m = from_pretrained_keras('ErnestBeckham/ViT-Lungs')
 
10
  hp['class_names'] = ["lung_aca", "lung_n", "lung_scc"]
11
 
12
  def main():
13
+ st.title("Lung Cancer Classification")
14
 
15
  # Upload image through drag and drop
16
  uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])