Update app.py

app.py

@@ -14,8 +14,8 @@ def cosine_similarity(vec1: List[int], vec2: List[int]):
     return 1 - spatial.distance.cosine(vec1, vec2)
     
 
-uae = AnglE.from_pretrained('WhereIsAI/UAE-Large-V1')
-uae_code = AnglE.from_pretrained('WhereIsAI/UAE-Code-Large-V1')
+uae = AnglE.from_pretrained('WhereIsAI/UAE-Large-V1').to('cpu')
+uae_code = AnglE.from_pretrained('WhereIsAI/UAE-Code-Large-V1').to('cpu')
 
 
 def code_similarity(code1, code2, model):