Spaces:
Sleeping
Sleeping
File size: 2,317 Bytes
691f2f2 0c1a34a 691f2f2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
import streamlit as st
from groq import Groq
import time
import os
# Function to analyze an image from a URL with retry mechanism
def analyze_image(image_url, retries=3, delay=2):
for attempt in range(retries):
try:
client = Groq()
completion = client.chat.completions.create(
model="llava-v1.5-7b-4096-preview",
messages=[
{
"role": "user",
"content": [
{"type": "text", "text": "What's in this image?"},
{"type": "image_url", "image_url": {"url": image_url}},
]
}
],
temperature=1,
max_tokens=1024,
top_p=1,
stream=False,
stop=None,
)
return completion.choices[0].message.content
except Exception as e:
if attempt < retries - 1:
st.error(f"API issue encountered: {e}. Retrying in {delay} seconds...")
else:
return f"Failed after {retries} attempts. Error: {e}"
time.sleep(delay) # Wait before retrying
# Streamlit app
st.title("Image Analyzer with Groq")
st.write(
"Type image url below and Groq will describe the image! "
"To use this app, you need to provide an Groq API key, which you can get [here](https://console.groq.com/keys). "
)
st.write("Enter an image URL to describe the image.")
model_options = [
"llava-v1.5-7b-4096-preview",
"llama-3.2-1b-preview",
"llama-3.2-3b-preview",
]
with st.sidebar:
selected_model = st.selectbox("Select any Groq Model", model_options)
groq_api_key = st.text_input("Groq API Key", type="password")
if not groq_api_key:
st.info("Please add your Groq API key to continue.", icon="🗝️")
else:
# Set it as an environment variable
os.environ["GROQ_API_KEY"] = groq_api_key
# Main section
try:
image_url = st.text_input("Enter Image URL")
if image_url:
ai_response = analyze_image(image_url)
st.image(image_url, use_column_width=True)
st.write("AI's Response:", ai_response)
except Exception as e:
st.error(f"API issue encountered: {e}.")
|