ifire commited on
Commit
f3d166c
1 Parent(s): cdd7dd3

Add hallways.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -81,7 +81,7 @@ def prompt_to_layout(user_prompt, intensity, fpath=None):
81
  if(containsNumber(user_prompt) == True):
82
  spaced_prompt = user_prompt.split(' ')
83
  new_prompt = ' '.join([word if word.isdigit() == False else num2words(int(word)).lower() for word in spaced_prompt])
84
- model_prompt = '[User prompt] Include hallways when rooms require. {} [Layout]'.format(new_prompt)
85
  top_p, top_k = creativity(intensity)
86
  model_prompt = '[User prompt] {} [Layout]'.format(user_prompt)
87
  input_ids = tokenizer(model_prompt, return_tensors='pt').to(device)
 
81
  if(containsNumber(user_prompt) == True):
82
  spaced_prompt = user_prompt.split(' ')
83
  new_prompt = ' '.join([word if word.isdigit() == False else num2words(int(word)).lower() for word in spaced_prompt])
84
+ model_prompt = '[User prompt] Hallways are adjacent to bedrooms. {} [Layout]'.format(new_prompt)
85
  top_p, top_k = creativity(intensity)
86
  model_prompt = '[User prompt] {} [Layout]'.format(user_prompt)
87
  input_ids = tokenizer(model_prompt, return_tensors='pt').to(device)