Refactor generate3d function in inference.py to enhance readability by removing unnecessary blank lines. Update requirements.txt to specify versions for gradio and huggingface-hub for better dependency management.

inference.py

@@ -7,8 +7,8 @@ import tempfile
 from mesh import Mesh
 import zipfile
 from util.renderer import Renderer
-
 def generate3d(model, rgb, ccm, device):
+
     model.renderer = Renderer(tet_grid_size=model.tet_grid_size, camera_angle_num=model.camera_angle_num,
                                 scale=model.input.scale, geo_type = model.geo_type)
 
@@ -17,6 +17,7 @@ def generate3d(model, rgb, ccm, device):
     color = color_tri.permute(2,0,1)
     xyz = xyz_tri.permute(2,0,1)
 
+
     def get_imgs(color):
         # color : [C, H, W*6]
         color_list = []
@@ -37,6 +38,7 @@ def generate3d(model, rgb, ccm, device):
     # 3D visualize
     model.eval()
     
+
     if model.denoising == True:
         tnew = 20
         tnew = torch.randint(tnew, tnew+1, [triplane.shape[0]], dtype=torch.long, device=triplane.device)
@@ -51,6 +53,7 @@ def generate3d(model, rgb, ccm, device):
     else:
         triplane_feature2 = model.unet2(triplane)
         
+
     with torch.no_grad():
         data_config = {
             'resolution': [1024, 1024],
@@ -62,6 +65,7 @@ def generate3d(model, rgb, ccm, device):
         data_config['verts'] = verts[0]
         data_config['faces'] = faces
         
+
     from kiui.mesh_utils import clean_mesh
     verts, faces = clean_mesh(data_config['verts'].squeeze().cpu().numpy().astype(np.float32), data_config['faces'].squeeze().cpu().numpy().astype(np.int32), repair = False, remesh=True, remesh_size=0.005, remesh_iters=1)
     data_config['verts'] = torch.from_numpy(verts).cuda().contiguous()

requirements.txt

@@ -1,5 +1,5 @@
-gradio
-huggingface-hub
+gradio==4.21.0
+huggingface-hub==0.16.4
 diffusers==0.24.0
 einops==0.7.0
 Pillow==10.1.0