MarcusLoren commited on
Commit
90f1226
1 Parent(s): 5e3add1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import torch
 
2
  from meshgpt_pytorch import (
3
  MeshTransformer,
4
  mesh_render
@@ -9,6 +10,7 @@ import gradio as gr
9
  import numpy as np
10
 
11
  transformer = MeshTransformer.from_pretrained("MarcusLoren/MeshGPT-preview")
 
12
 
13
  def save_as_obj(file_path):
14
  v, f = igl.read_triangle_mesh(file_path)
@@ -22,6 +24,10 @@ def predict(text, num_input, num_temp):
22
  transformer.eval()
23
  labels = [label.strip() for label in text.split(',')]
24
  output = []
 
 
 
 
25
  if num_input > 1:
26
  for label in labels:
27
  output.append((transformer.generate(texts = [label ] * num_input, temperature = num_temp)))
 
1
  import torch
2
+ import time
3
  from meshgpt_pytorch import (
4
  MeshTransformer,
5
  mesh_render
 
10
  import numpy as np
11
 
12
  transformer = MeshTransformer.from_pretrained("MarcusLoren/MeshGPT-preview")
13
+ total_generations = 0
14
 
15
  def save_as_obj(file_path):
16
  v, f = igl.read_triangle_mesh(file_path)
 
24
  transformer.eval()
25
  labels = [label.strip() for label in text.split(',')]
26
  output = []
27
+ total_generations +=1
28
+ current_time = time.time()
29
+ formatted_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(current_time))
30
+ print(formatted_time, " Input:", text, "num_input", num_input, "num_temp",num_temp , "total_generations", total_generations)
31
  if num_input > 1:
32
  for label in labels:
33
  output.append((transformer.generate(texts = [label ] * num_input, temperature = num_temp)))