Arjs commited on
Commit
445e7b5
·
verified ·
1 Parent(s): dda77cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -13
app.py CHANGED
@@ -8,21 +8,10 @@ from langchain_groq import ChatGroq
8
  import os
9
 
10
  # Load BLIP model and processor
11
- #processor = AutoProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
12
- #model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
13
 
14
 
15
- import requests
16
-
17
- API_URL = "https://api-inference.huggingface.co/models/Salesforce/blip2-opt-2.7b"
18
- headers = {"Authorization": "Bearer hf_xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
19
-
20
- def query(filename):
21
- with open(filename, "rb") as f:
22
- data = f.read()
23
- response = requests.post(API_URL, headers=headers, data=data)
24
- return response.json()
25
-
26
 
27
  # Define the image description function
28
  def image_description(image):
 
8
  import os
9
 
10
  # Load BLIP model and processor
11
+ processor = AutoProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
12
+ model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
13
 
14
 
 
 
 
 
 
 
 
 
 
 
 
15
 
16
  # Define the image description function
17
  def image_description(image):