neha-ai-playground commited on
Commit
5e250bb
·
1 Parent(s): 0c9d330

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -1,13 +1,17 @@
1
  import os
2
  from pathlib import Path
3
  import streamlit as st
4
- from datasets import load_dataset
5
  from dotenv import load_dotenv
6
 
7
  if Path(".env").is_file():
8
  load_dotenv(".env")
9
-
10
  st.set_page_config(layout="wide")
11
-
12
  HF_TOKEN = os.getenv("HF_TOKEN")
13
- st.title(HF_TOKEN)
 
 
 
 
 
 
 
1
  import os
2
  from pathlib import Path
3
  import streamlit as st
4
+ from transformers import pipeline
5
  from dotenv import load_dotenv
6
 
7
  if Path(".env").is_file():
8
  load_dotenv(".env")
 
9
  st.set_page_config(layout="wide")
 
10
  HF_TOKEN = os.getenv("HF_TOKEN")
11
+
12
+ def img2Text(url)
13
+ image_to_text = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large")
14
+ text = image_to_text(url)
15
+ print(text)
16
+ return text
17
+ img2Text(photo.png)