JiantaoLin commited on
Commit
da20481
·
1 Parent(s): 671daef
Files changed (2) hide show
  1. app.py +1 -1
  2. 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
- @spaces.GPU
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: