dkatz2391 commited on
Commit
df09626
·
verified ·
1 Parent(s): 31d67e5

fix endpont state object

Browse files
Files changed (1) hide show
  1. app.py +21 -1
app.py CHANGED
@@ -271,4 +271,24 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
271
  if __name__ == "__main__":
272
  pipeline = TrellisTextTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-text-xlarge")
273
  pipeline.cuda()
274
- demo.launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
271
  if __name__ == "__main__":
272
  pipeline = TrellisTextTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-text-xlarge")
273
  pipeline.cuda()
274
+ demo.launch()
275
+
276
+ # --- API-only endpoint for server integration ---
277
+ # This exposes text_to_3d with gr.JSON() as the first output, so the state object is included in the API response.
278
+ # Not wired to the UI; use for API calls only.
279
+ api_text_to_3d = gr.Interface(
280
+ fn=lambda prompt, seed, ss_guidance_strength, ss_sampling_steps, slat_guidance_strength, slat_sampling_steps: text_to_3d(
281
+ prompt, seed, ss_guidance_strength, ss_sampling_steps, slat_guidance_strength, slat_sampling_steps, gr.Request()
282
+ ),
283
+ inputs=[
284
+ gr.Textbox(label="Text Prompt"),
285
+ gr.Number(label="Seed"),
286
+ gr.Number(label="SS Guidance Strength"),
287
+ gr.Number(label="SS Sampling Steps"),
288
+ gr.Number(label="SLAT Guidance Strength"),
289
+ gr.Number(label="SLAT Sampling Steps"),
290
+ ],
291
+ outputs=[gr.JSON(label="State Object"), gr.Textbox(label="Video Path")],
292
+ allow_flagging="never",
293
+ description="API endpoint for text_to_3d that returns the state object as JSON. Not for UI use.",
294
+ )