NoorAhmadHaral commited on
Commit
64f5667
·
verified ·
1 Parent(s): ab7a459

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -2,8 +2,11 @@ import os
2
  import gradio as gr
3
  from transformers import pipeline
4
 
5
- pipe = pipeline("image-to-text",
6
- model="./models/Salesforce/blip-image-captioning-base")
 
 
 
7
 
8
 
9
  def launch(input):
 
2
  import gradio as gr
3
  from transformers import pipeline
4
 
5
+ # pipe = pipeline("image-to-text",
6
+ # model="./models/Salesforce/blip-image-captioning-base")
7
+
8
+ pipe = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
9
+
10
 
11
 
12
  def launch(input):