Update demo_model.py

demo_model.py

@@ -79,7 +79,7 @@ class LGGMText2Graph_Demo(pl.LightningModule):
         with torch.no_grad():
             prompt_emb = self.text_encoder(**encoded_input).hidden_states[-1][:, 0]
 
-        samples = self.sample_batch(5, cond_emb = prompt_emb.to(self.device), num_nodes = num_nodes)
+        samples = self.sample_batch(3, cond_emb = prompt_emb.to(self.device), num_nodes = num_nodes)
 
         nx_graphs = []
         for graph in samples: