mrbeliever commited on
Commit
986b8bd
·
verified ·
1 Parent(s): 8aa1ea9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ def generate_caption(image_base64, api_key):
26
  api_url = "https://api.studio.nebius.ai/v1/chat/completions"
27
  headers = {"Authorization": f"Bearer {api_key}"}
28
  payload = {
29
- "model": "Qwen/Qwen2-VL-7B-Instruct",
30
  "messages": [
31
  {"role": "system", "content": "You are an image-to-text prompt converter for AI image generation."},
32
  {"role": "user", "content": [
 
26
  api_url = "https://api.studio.nebius.ai/v1/chat/completions"
27
  headers = {"Authorization": f"Bearer {api_key}"}
28
  payload = {
29
+ "model": "mistralai/Mistral-Small-3.1-24B-Instruct-2503",
30
  "messages": [
31
  {"role": "system", "content": "You are an image-to-text prompt converter for AI image generation."},
32
  {"role": "user", "content": [