brucewayne0459 commited on
Commit
fccc04f
1 Parent(s): ef49fac

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -5
README.md CHANGED
@@ -1,5 +1,17 @@
1
  ---
2
  license: apache-2.0
 
 
 
 
 
 
 
 
 
 
 
 
3
  ---
4
  # Model Card for PaliGemma Dermatology Model
5
 
@@ -11,9 +23,8 @@ This model, based on the PaliGemma-3B architecture, has been fine-tuned for derm
11
 
12
 
13
  - **Developed by:** Bruce_Wayne
14
- - **Funded by [optional]:** Jhonny and koti
15
  - **Model type:** vision model
16
- - **Finetuned from model [optional]:** https://huggingface.co/google/paligemma-3b-pt-224
17
  - **LoRa Adaptors used:** Yes
18
  - **Intended use:** Medical image analysis, specifically for dermatology
19
  **
@@ -30,14 +41,14 @@ The model can be directly used for analyzing dermatology images, providing insig
30
 
31
  ## How to Get Started with the Model
32
 
33
- ** python
34
 
35
  from transformers import AutoProcessor, PaliGemmaForConditionalGeneration
36
 
37
  model_id = "brucewayne0459/paligemma_derm"
38
  processor = AutoProcessor.from_pretrained(model_id)
39
  model = PaliGemmaForConditionalGeneration.from_pretrained(model_id)
40
- **
41
  ## Training Details
42
 
43
  ### Training Data
@@ -97,4 +108,4 @@ The model achieved a final validation loss of approximately 0.2214, indicating r
97
 
98
  - **GPU:** 1xL4 GPU
99
  ## Model Card Authors
100
- Bruce_Wayne
 
1
  ---
2
  license: apache-2.0
3
+ datasets:
4
+ - joshuachou/SkinCAP
5
+ - HemanthKumarK/SKINgpt
6
+ language:
7
+ - en
8
+ pipeline_tag: image-to-text
9
+ tags:
10
+ - biology
11
+ - skin
12
+ - skin disease
13
+ - cancer
14
+ - medical
15
  ---
16
  # Model Card for PaliGemma Dermatology Model
17
 
 
23
 
24
 
25
  - **Developed by:** Bruce_Wayne
 
26
  - **Model type:** vision model
27
+ - **Finetuned from model:** https://huggingface.co/google/paligemma-3b-pt-224
28
  - **LoRa Adaptors used:** Yes
29
  - **Intended use:** Medical image analysis, specifically for dermatology
30
  **
 
41
 
42
  ## How to Get Started with the Model
43
 
44
+ ```python
45
 
46
  from transformers import AutoProcessor, PaliGemmaForConditionalGeneration
47
 
48
  model_id = "brucewayne0459/paligemma_derm"
49
  processor = AutoProcessor.from_pretrained(model_id)
50
  model = PaliGemmaForConditionalGeneration.from_pretrained(model_id)
51
+ ```
52
  ## Training Details
53
 
54
  ### Training Data
 
108
 
109
  - **GPU:** 1xL4 GPU
110
  ## Model Card Authors
111
+ Bruce_Wayne