JiantaoLin
commited on
Commit
·
da20481
1
Parent(s):
671daef
new
Browse files- app.py +1 -1
- pipeline/kiss3d_wrapper.py +1 -0
app.py
CHANGED
@@ -284,7 +284,7 @@ def image_to_base64(image_path):
|
|
284 |
"""Converts an image file to a base64-encoded string."""
|
285 |
with open(image_path, "rb") as img_file:
|
286 |
return base64.b64encode(img_file.read()).decode('utf-8')
|
287 |
-
|
288 |
def main():
|
289 |
|
290 |
torch.set_grad_enabled(False)
|
|
|
284 |
"""Converts an image file to a base64-encoded string."""
|
285 |
with open(image_path, "rb") as img_file:
|
286 |
return base64.b64encode(img_file.read()).decode('utf-8')
|
287 |
+
|
288 |
def main():
|
289 |
|
290 |
torch.set_grad_enabled(False)
|
pipeline/kiss3d_wrapper.py
CHANGED
@@ -256,6 +256,7 @@ class kiss3d_wrapper(object):
|
|
256 |
caption_text = self.get_detailed_prompt(caption_text)
|
257 |
|
258 |
return caption_text
|
|
|
259 |
@spaces.GPU
|
260 |
def get_detailed_prompt(self, prompt, seed=None):
|
261 |
if self.llm_model is not None:
|
|
|
256 |
caption_text = self.get_detailed_prompt(caption_text)
|
257 |
|
258 |
return caption_text
|
259 |
+
|
260 |
@spaces.GPU
|
261 |
def get_detailed_prompt(self, prompt, seed=None):
|
262 |
if self.llm_model is not None:
|