yetessam commited on
Commit
eff6977
·
verified ·
1 Parent(s): 12463d7

Adding agent type and compute strings

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -59,11 +59,21 @@ def main():
59
  # We want to create a basic GUI calling the ContentAgentUI..
60
  # The default flow should be to show the "Wake Endpoint" button
61
  # If the end point is ready, don't show it.
 
 
 
 
 
 
62
  ui = ContentAgentUI(
 
63
  endpoint_uri=uri,
64
  is_healthy=ok,
65
  health_message=msg,
66
  agent_initializer=initialize_agent,
 
 
 
67
  )
68
  ui.launch()
69
 
 
59
  # We want to create a basic GUI calling the ContentAgentUI..
60
  # The default flow should be to show the "Wake Endpoint" button
61
  # If the end point is ready, don't show it.
62
+
63
+
64
+ agent_type = """ deepseek-ai/DeepSeek-R1-Distill-Qwen-32B` text generation model """
65
+ compute = "GCP · Nvidia L4 · 4x GPUs · 96 GB"
66
+
67
+
68
  ui = ContentAgentUI(
69
+
70
  endpoint_uri=uri,
71
  is_healthy=ok,
72
  health_message=msg,
73
  agent_initializer=initialize_agent,
74
+ agent_type = agent_type,
75
+ compute = compute
76
+
77
  )
78
  ui.launch()
79