diff --git a/app/fastvlm.py b/app/fastvlm.py index 0ae5b90..b864f9f 100644 --- a/app/fastvlm.py +++ b/app/fastvlm.py @@ -87,15 +87,20 @@ class FastVLM: 'timecost': t2 - t1, 'content': outputs } + debug(f'Exception happened .......') + return None async def generate(self, image_file, prompt): f = awaitify(self._generate) - return await f(image_file, promot) + return await f(image_file, prompt) +fastvlm = None def init(): + global fastvlm g = ServerEnv() - k = FastVLM() - g.generate = k.generate + g.fastvlm = fastvlm + fastvlm = FastVLM() + g.generate = fastvlm.generate if __name__ == "__main__": webapp(init) diff --git a/wwwroot/v1/generate/index.dspy b/wwwroot/v1/generate/index.dspy index ec3dcdc..874009d 100644 --- a/wwwroot/v1/generate/index.dspy +++ b/wwwroot/v1/generate/index.dspy @@ -3,5 +3,6 @@ image_path = realpath(params_kw.image_path) prompt = params_kw.prompt debug(f'{image_path=}, {prompt=}') d = await generate(image_path, prompt) +debug(f'{image_path=}, {prompt=}, {d=}') return d