Update app.py

app.py

@@ -64,7 +64,7 @@ def next_image():
             response = requests.get(output['file1'])
             image = Image.open(BytesIO(response.content))
 
-            embs.append(torch.tensor([float(i) for i in urlopen(output['file2']).read().decode('utf-8').split(', ')]))
+            embs.append(torch.tensor([float(i) for i in urlopen(output['file2']).read().decode('utf-8').split(', ')]).unsqueeze(0))
             return image
         else:
             print('######### Roaming #########')
@@ -105,7 +105,7 @@ def next_image():
             image = Image.open(BytesIO(response.content))
 
 
-            im_emb = torch.tensor([float(i) for i in urlopen(output['file2']).read().decode('utf-8').split(', ')])
+            im_emb = torch.tensor([float(i) for i in urlopen(output['file2']).read().decode('utf-8').split(', ')]).unsqueeze(0)
             embs.append(im_emb)
 
             torch.save(lin_class.coef_, f'./{start_time}.pt')