Spaces:
				
			
			
	
			
			
					
		Running
		
	
	
	
			
			
	
	
	
	
		
		
					
		Running
		
	Update app.py
Browse files
    	
        app.py
    CHANGED
    
    | @@ -28,7 +28,7 @@ import gradio as gr | |
| 28 |  | 
| 29 | 
             
            device = 'cpu'
         | 
| 30 | 
             
            le = LabelEncoder()
         | 
| 31 | 
            -
            le = joblib.load(" | 
| 32 |  | 
| 33 | 
             
            class ModelPre(torch.nn.Module):
         | 
| 34 | 
             
                def __init__(self):
         | 
| @@ -44,7 +44,7 @@ class ModelPre(torch.nn.Module): | |
| 44 | 
             
                def forward(self, data):
         | 
| 45 | 
             
                    return self.embedding(data)
         | 
| 46 |  | 
| 47 | 
            -
            model = torch.load(" | 
| 48 |  | 
| 49 | 
             
            modelm = ModelPre()
         | 
| 50 | 
             
            modelm.load_state_dict(model['model'])
         | 
|  | |
| 28 |  | 
| 29 | 
             
            device = 'cpu'
         | 
| 30 | 
             
            le = LabelEncoder()
         | 
| 31 | 
            +
            le = joblib.load("~/SVD/le.gz")
         | 
| 32 |  | 
| 33 | 
             
            class ModelPre(torch.nn.Module):
         | 
| 34 | 
             
                def __init__(self):
         | 
|  | |
| 44 | 
             
                def forward(self, data):
         | 
| 45 | 
             
                    return self.embedding(data)
         | 
| 46 |  | 
| 47 | 
            +
            model = torch.load("~/SVD/GeoG.pth", map_location=torch.device(device))
         | 
| 48 |  | 
| 49 | 
             
            modelm = ModelPre()
         | 
| 50 | 
             
            modelm.load_state_dict(model['model'])
         |