bstraehle commited on
Commit
d17956e
·
verified ·
1 Parent(s): 80948ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -14
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  import base64, os, threading
3
 
4
- from openai import OpenAI
5
 
6
  lock = threading.Lock()
7
 
@@ -29,22 +29,22 @@ def invoke(openai_api_key, prompt, image):
29
  content = ""
30
 
31
  try:
32
- client = OpenAI()
33
 
34
- img_b64 = get_img_b64(image)
35
 
36
- completion = client.chat.completions.create(
37
- max_tokens = config["max_tokens"],
38
- messages = [{"role": "user",
39
- "content": [{"type": "text",
40
- "text": prompt},
41
- {"type": "image_url",
42
- "image_url": {"url": f"data:image/jpeg;base64,{img_b64}"}}]}],
43
- model = config["model"],
44
- temperature = config["temperature"]
45
- )
46
 
47
- content = completion.choices[0].message.content
48
 
49
  print("###")
50
  print(content)
 
1
  import gradio as gr
2
  import base64, os, threading
3
 
4
+ #from openai import OpenAI
5
 
6
  lock = threading.Lock()
7
 
 
29
  content = ""
30
 
31
  try:
32
+ #client = OpenAI()
33
 
34
+ #img_b64 = get_img_b64(image)
35
 
36
+ #completion = client.chat.completions.create(
37
+ # max_tokens = config["max_tokens"],
38
+ # messages = [{"role": "user",
39
+ # "content": [{"type": "text",
40
+ # "text": prompt},
41
+ # {"type": "image_url",
42
+ # "image_url": {"url": f"data:image/jpeg;base64,{img_b64}"}}]}],
43
+ # model = config["model"],
44
+ # temperature = config["temperature"]
45
+ #)
46
 
47
+ #content = completion.choices[0].message.content
48
 
49
  print("###")
50
  print(content)