Spaces:

IamVicky111
/

Object_Detection

Sleeping

IamVicky111 commited on Jun 12, 2024

Commit

51be3dc

verified ·

1 Parent(s): 4bfd9de

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,8 +1,9 @@
 import os
 from transformers import pipeline
 pipe = pipeline("image-to-text",
-                model="./models/Salesforce/blip-image-captioning-base")
 def launch(input):
     out = pipe(input)

 import os
+import gradio as gr
 from transformers import pipeline
 pipe = pipeline("image-to-text",
+                model="Salesforce/blip-image-captioning-base")
 def launch(input):
     out = pipe(input)