Upload 14 files

.env

@@ -0,0 +1,2 @@
+HF_HOME='./huggingface_folder'
+HF_HUB_CACHE='./huggingface_folder/hub'

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+huggingface_folder/hub/models--Ajay-user--vit-base-patch16-224-finetuned-pneumonia-detection/blobs/cf194b576e2d571995cc9c1cbe99a5ba9e80e0023d2a22e43f8abf035560b002 filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,93 @@
+
+import streamlit as st
+import pandas as pd
+import requests
+from dotenv import load_dotenv
+
+from transformers import pipeline
+from PIL import Image
+
+from info import pneumonia, covid19, vit_base_patch_16
+
+load_dotenv()
+
+URL = 'https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcTcY7VeTAy72aEPJbHmABvnGzW5gzrvSKRzOg&usqp=CAU'
+
+
+def download_image():
+    if st.session_state.img_url:
+        st.session_state['image'] = Image.open(
+            requests.get(st.session_state.img_url, stream=True).raw)
+    else:
+        del st.session_state['image']
+
+
+def file_upload():
+    if st.session_state.file_upload:
+        st.session_state['image'] = Image.open(st.session_state.file_upload)
+    else:
+        del st.session_state['image']
+
+
+def cam_upload():
+    if st.session_state.camera:
+        st.session_state['image'] = st.session_state.camera
+    else:
+        del st.session_state['image']
+
+
+if 'image' not in st.session_state:
+    st.session_state['image'] = Image.open(requests.get(URL, stream=True).raw)
+
+st.header("Pneumonia and Covid19 Detector")
+
+with st.sidebar:
+    img_upload, cam_upload, url_upload = st.tabs(
+        ['📂 Upload', '📸 CAMERA', '🔗 URL'])
+
+    with img_upload:
+        uploaded_img = st.file_uploader(
+            label="Upload an X-ray image", on_change=file_upload, key='file_upload'
+        )
+    with cam_upload:
+        camera_img = st.camera_input(
+            label='Take a picture of X-ray', on_change=cam_upload, key='camera'
+        )
+    with url_upload:
+        img_url = st.text_input(
+            label="Enter the X-ray URL", value=URL, on_change=download_image, key="img_url"
+        )
+
+    st.image(st.session_state.image)
+
+    analyze_btn = st.button(label='Analyze X-ray', type='primary',
+                            use_container_width=True, key='analyze_btn')
+
+if st.session_state.image and st.session_state.analyze_btn:
+    with st.spinner():
+        pipe = pipeline("image-classification",
+                        model="Ajay-user/vit-base-patch16-224-finetuned-pneumonia-detection")
+        response = pipe(st.session_state.image)
+
+        df = pd.DataFrame(response)
+        result = df.nlargest(n=1, columns='score')
+        result_body = f'Model predicts : {result["label"].item()} with {result["score"].item()*100 :0.2f}% confidence'
+
+        with st.expander(label=result_body, expanded=True):
+            st.subheader(body=f':red[{result["label"].item()}] Detected')
+            st.bar_chart(data=df, x='label', y='score')
+
+        with st.expander(label="X-ray image analyzed"):
+            st.image(st.session_state.image)
+
+        with st.expander(label="Model Details"):
+            st.markdown(body=vit_base_patch_16)
+
+else:
+    tab_1, tab_2 = st.tabs(['Pneumonia', 'Coronavirus'])
+    with tab_1:
+        st.subheader('Pneumonia')
+        st.markdown(body=pneumonia)
+    with tab_2:
+        st.subheader('Coronavirus')
+        st.markdown(body=covid19)

huggingface_folder/hub/models--Ajay-user--vit-base-patch16-224-finetuned-pneumonia-detection/blobs/273d24924a8bd10227b532d603484f165dcb0b5d

@@ -0,0 +1,22 @@
+{
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "image_processor_type": "ViTImageProcessor",
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "height": 224,
+    "width": 224
+  }
+}

huggingface_folder/hub/models--Ajay-user--vit-base-patch16-224-finetuned-pneumonia-detection/blobs/cf194b576e2d571995cc9c1cbe99a5ba9e80e0023d2a22e43f8abf035560b002

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cf194b576e2d571995cc9c1cbe99a5ba9e80e0023d2a22e43f8abf035560b002
+size 343227052

huggingface_folder/hub/models--Ajay-user--vit-base-patch16-224-finetuned-pneumonia-detection/blobs/f01e5e4a53624256f48d3bc82b547e67ee5dda49

@@ -0,0 +1,34 @@
+{
+  "_name_or_path": "google/vit-base-patch16-224",
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "COVID19",
+    "1": "NORMAL",
+    "2": "PNEUMONIA"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "COVID19": 0,
+    "NORMAL": 1,
+    "PNEUMONIA": 2
+  },
+  "layer_norm_eps": 1e-12,
+  "model_type": "vit",
+  "num_attention_heads": 12,
+  "num_channels": 3,
+  "num_hidden_layers": 12,
+  "patch_size": 16,
+  "problem_type": "single_label_classification",
+  "qkv_bias": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.38.2"
+}

huggingface_folder/hub/models--Ajay-user--vit-base-patch16-224-finetuned-pneumonia-detection/refs/main

@@ -0,0 +1 @@
+d301c243e1d96c5531bc56dd8c49f4aeb1cdc0d6

huggingface_folder/hub/version.txt

@@ -0,0 +1 @@
+1

info.py

@@ -0,0 +1,122 @@
+pneumonia = '''
+Pneumonia is an infection that inflames the air sacs in one or both lungs. 
+The air sacs may fill with fluid or pus (purulent material), 
+causing cough with phlegm or pus, fever, chills, and difficulty breathing. 
+A variety of organisms, including bacteria, viruses and fungi, can cause pneumonia.
+
+Pneumonia can range in seriousness from mild to life-threatening. 
+It is most serious for infants and young children, people older than age 65, 
+and people with health problems or weakened immune systems.
+
+#### Symptoms
+The signs and symptoms of pneumonia vary from mild to severe, 
+depending on factors such as the type of germ causing the infection, 
+and your age and overall health. 
+Mild signs and symptoms often are similar to those of a cold or flu, but they last longer.
+
+#### Signs and symptoms of pneumonia may include:
+
+* Chest pain when you breathe or cough
+* Confusion or changes in mental awareness (in adults age 65 and older)
+* Cough, which may produce phlegm
+* Fatigue
+* Fever, sweating and shaking chills
+* Lower than normal body temperature (in adults older than age 65 and people with weak immune systems)
+* Nausea, vomiting or diarrhea
+* Shortness of breath
+
+Newborns and infants may not show any sign of the infection. 
+Or they may vomit, have a fever and cough, 
+appear restless or tired and without energy, or have difficulty breathing and eating.
+
+#### When to see a doctor
+See your doctor if you have difficulty breathing, 
+chest pain, persistent fever of 102 F (39 C) or higher, or persistent cough, 
+especially if you're coughing up pus.
+
+#### It's especially important that people in these high-risk groups see a doctor:
+
+* Adults older than age 65
+* Children younger than age 2 with signs and symptoms
+* People with an underlying health condition or weakened immune system
+* People receiving chemotherapy or taking medication that suppresses the immune system
+
+For some older adults and people with heart failure or chronic lung problems, pneumonia can quickly become a life-threatening condition.
+'''
+
+
+covid19 = '''
+
+Coronavirus disease (COVID-19) is an infectious disease caused by the SARS-CoV-2 virus.
+
+Most people infected with the virus will experience mild to moderate respiratory illness and recover without requiring special treatment. However, some will become seriously ill and require medical attention. Older people and those with underlying medical conditions like cardiovascular disease, diabetes, chronic respiratory disease, or cancer are more likely to develop serious illness. Anyone can get sick with COVID-19 and become seriously ill or die at any age. 
+
+The best way to prevent and slow down transmission is to be well informed about the disease and how the virus spreads. Protect yourself and others from infection by staying at least 1 metre apart from others, wearing a properly fitted mask, and washing your hands or using an alcohol-based rub frequently. Get vaccinated when it’s your turn and follow local guidance.
+
+The virus can spread from an infected person’s mouth or nose in small liquid particles when they cough, sneeze, speak, sing or breathe. These particles range from larger respiratory droplets to smaller aerosols. It is important to practice respiratory etiquette, for example by coughing into a flexed elbow, and to stay home and self-isolate until you recover if you feel unwell.
+
+#### Symptoms of COVID-19
+* Fever or chills
+* Cough
+* Shortness of breath or difficulty breathing
+* Fatigue
+* Muscle or body aches
+* Headache
+* New loss of taste or smell
+* Sore throat
+* Congestion or runny nose
+* Nausea or vomiting
+* Diarrhea
+
+
+#### When to Seek Emergency Medical Attention
+**Look for emergency warning signs* for COVID 19:**
+
+* Trouble breathing
+* Persistent pain or pressure in the chest
+* New confusion
+* Inability to wake or stay awake
+* Pale, gray, or blue-colored skin, lips, or nail beds, depending on skin tone
+
+'''
+
+vit_base_patch_16 = '''
+
+# vit-base-patch16-224-finetuned-pneumonia-detection
+
+This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on the imagefolder dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.0938
+- Accuracy: 0.9728
+
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 5e-05
+- train_batch_size: 64
+- eval_batch_size: 64
+- seed: 42
+- gradient_accumulation_steps: 4
+- total_train_batch_size: 256
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: linear
+- lr_scheduler_warmup_ratio: 0.1
+- num_epochs: 3
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Accuracy |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 0.2082        | 0.99  | 20   | 0.1462          | 0.9402   |
+| 0.0832        | 1.98  | 40   | 0.0998          | 0.9658   |
+| 0.0517        | 2.96  | 60   | 0.0938          | 0.9728   |
+
+
+### Framework versions
+
+- Transformers 4.38.2
+- Pytorch 2.1.2
+- Datasets 2.18.0
+- Tokenizers 0.15.2
+'''

requirements.txt

@@ -0,0 +1,72 @@
+altair==5.2.0
+attrs==23.2.0
+blinker==1.7.0
+cachetools==5.3.3
+certifi==2024.2.2
+charset-normalizer==3.3.2
+click==8.1.7
+filelock==3.13.1
+fsspec==2024.2.0
+gitdb==4.0.11
+GitPython==3.1.42
+huggingface-hub==0.21.4
+idna==3.6
+importlib_metadata==7.0.2
+Jinja2==3.1.3
+jsonschema==4.21.1
+jsonschema-specifications==2023.12.1
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+mdurl==0.1.2
+mpmath==1.3.0
+networkx==3.2.1
+numpy==1.26.4
+nvidia-cublas-cu12==12.1.3.1
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cudnn-cu12==8.9.2.26
+nvidia-cufft-cu12==11.0.2.54
+nvidia-curand-cu12==10.3.2.106
+nvidia-cusolver-cu12==11.4.5.107
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-nccl-cu12==2.19.3
+nvidia-nvjitlink-cu12==12.4.99
+nvidia-nvtx-cu12==12.1.105
+packaging==23.2
+pandas==2.2.1
+pillow==10.2.0
+protobuf==4.25.3
+pyarrow==15.0.0
+pydeck==0.8.1b0
+Pygments==2.17.2
+python-dateutil==2.9.0.post0
+python-dotenv==1.0.1
+pytz==2024.1
+PyYAML==6.0.1
+referencing==0.33.0
+regex==2023.12.25
+requests==2.31.0
+rich==13.7.1
+rpds-py==0.18.0
+safetensors==0.4.2
+six==1.16.0
+smmap==5.0.1
+streamlit==1.31.1
+sympy==1.12
+tenacity==8.2.3
+tokenizers==0.15.2
+toml==0.10.2
+toolz==0.12.1
+torch==2.2.1
+tornado==6.4
+tqdm==4.66.2
+transformers==4.38.2
+triton==2.2.0
+typing_extensions==4.10.0
+tzdata==2024.1
+tzlocal==5.2
+urllib3==2.2.1
+validators==0.22.0
+watchdog==4.0.0
+zipp==3.17.0