Spaces:
Sleeping
Sleeping
Add hallways.
Browse files
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]
|
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)
|