Hugging Face's logo

Spaces:
Shreeraj
/
Metal_Defects_Classification_Application
Runtime error

App Files Community
Shreeraj commited on
Commit
84a7d93
·
1 Parent(s): 93fcfa6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -34,8 +34,7 @@ inv_normalize = transforms.Normalize(
34
 
35
  # Gradio app interface
36
  def classify_image(inp, transperancy=0.8):
37
- #image = Image.fromarray((inp * 255).astype(np.uint8)) # Convert NumPy array to PIL Image
38
- #input_tensor = preprocess(image)
39
  input_tensor = preprocess(inp)
40
  input_batch = input_tensor.unsqueeze(0).to('cpu') # Create a batch
41
 
 
34
 
35
  # Gradio app interface
36
  def classify_image(inp, transperancy=0.8):
37
+ model.to("cpu")
 
38
  input_tensor = preprocess(inp)
39
  input_batch = input_tensor.unsqueeze(0).to('cpu') # Create a batch
40