Spaces:
Configuration error
Configuration error
File size: 3,366 Bytes
e500bb8 |
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 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 |
import openai
import requests
class API:
def __init__(self, host="0.0.0.0", port=8000):
self.prefix = "http://{}:{}/".format(host, port)
def post(self, endpoint, data):
return requests.post(self.prefix + endpoint, json=data).json()
class VFM_API(API):
def __init__(self, host='0.0.0.0', port=8123):
super().__init__(host, port)
def vqa(self, image_path: str, question: str = None):
if question is None:
question = "Describe the image:"
response = self.post("blip", {'image_path': image_path, 'question': question})
return response.get('response')
def controlnet(self, image_path: str, mask_path: str, prompt: str, **kwargs) -> str:
content = {"prompt": prompt, "image_path": image_path, "mask_image_path": mask_path}
content.update(kwargs)
response = self.post("controlnet", content).get('response') # return List[str]
response = response[0] if len(response) > 0 else "./static/images/NSFW.jpg" # NSFW
return response
def lineart(self, image_path: str, coarse=False, detect_resolution=768, image_resolution=768,
output_type="pil", **kwargs) -> str:
content = {"input_image": image_path, "coarse": coarse,
"detect_resolution": detect_resolution, "image_resolution": image_resolution,
"output_type": output_type}
content.update(kwargs)
return self.post('lineart', content).get('response')
class SSM_API(API):
def __init__(self, host='0.0.0.0', port=8123):
super().__init__(host, port)
def graph(self, image_path: str) -> str:
response = self.post('graph', {'image_path': image_path})
return response.get('response')
def dense(self, image_path: str) -> str:
response = self.post('densepose', {'image_path': image_path})
return response.get('response')
def segment(self, image_path: str, text_prompt: str = 'person',
box_threshold: float = 0.3, text_threshold: float = 0.25) -> str:
response = self.post('segment', {'image_path': image_path, 'text_prompt': text_prompt,
'box_threshold': box_threshold, 'text_threshold': text_threshold})
return response.get('response')
class CHAT_API:
def __init__(self, port: int = 8001, model="vicuna"):
super().__init__()
self.model = model
openai.api_base = f"http://localhost:{port}/v1"
openai.api_key = "EMPTY"
def chat(self, prompt, history, temperature=0.01, **kwargs):
history_ = []
for u, a in history:
history_.append({"role": 'user', "content": u if u is not None else ""})
history_.append({"role": 'assistant', "content": a if a is not None else ""})
history_.append({"role": "user", "content": prompt})
completion = openai.ChatCompletion.create(model=self.model, messages=history_, temperature=temperature,
**kwargs)
response = completion.choices[0].message.content
history_.append({"role": "assistant", "content": response})
history__ = []
for i in range(0, len(history_), 2):
history__.append((history_[i]["content"], history_[i + 1]["content"]))
return response, history
|