damerajee commited on
Commit
0796213
·
verified ·
1 Parent(s): a6bbbf5

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +34 -0
README.md ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ datasets:
4
+ - damerajee/clean_vqa_prt2
5
+ language:
6
+ - en
7
+ library_name: transformers
8
+ tags:
9
+ - 'small vision language model '
10
+ ---
11
+ # GPT-Vision-1-ft
12
+ This model is a finetuned version of [GPT-Vision-1](https://huggingface.co/damerajee/GPT-Vision)
13
+
14
+ # Training Details
15
+ * The model was trained for 1 epoch on 200k rows
16
+ * The Fine-tuning took about 8 hours
17
+
18
+ # Inference
19
+ ```python
20
+ from transformers import AutoModelForCausalLM
21
+ from PIL import Image
22
+
23
+ model = AutoModelForCausalLM.from_pretrained("damerajee/GPT-Vision-1-ft", trust_remote_code=True)
24
+
25
+ image_path = "Your_image_path"
26
+ image = Image.open(image_path)
27
+ image = image.convert('RGB')
28
+
29
+ question = "your_Quesition"
30
+ answer = model.generate(image=image,question=question,max_new_tokens=40)
31
+ print(answer)
32
+ ```
33
+
34
+ # Examples