cedpsam commited on
Commit
e8e197f
·
verified ·
1 Parent(s): 86ad2fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -4,7 +4,12 @@ from urllib.parse import quote
4
  from huggingface_hub import InferenceClient
5
  import numpy as np
6
  import gradio as gr
7
- client = InferenceClient("Salesforce/blip-image-captioning-base")
 
 
 
 
 
8
  def generate_img(prompt):
9
 
10
 
@@ -17,7 +22,7 @@ def pollinations_url_seedless(a, width=512, height=512):
17
 
18
  def interrogate(img):
19
 
20
- result = client.image_to_text(img)
21
  return result['generated_text']
22
  def rountrip(img):
23
  prompt=interrogate(img)
 
4
  from huggingface_hub import InferenceClient
5
  import numpy as np
6
  import gradio as gr
7
+
8
+
9
+ # Use a pipeline as a high-level helper
10
+ from transformers import pipeline
11
+
12
+ pipe = pipeline("image-to-text", model="genta-tech/nougat-base-onnx-quantized")
13
  def generate_img(prompt):
14
 
15
 
 
22
 
23
  def interrogate(img):
24
 
25
+ result = pipe(img)
26
  return result['generated_text']
27
  def rountrip(img):
28
  prompt=interrogate(img)