Spaces:
Running
Running
Update app.py
Browse files
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)))
|