Fix Error : Add integration of gpu device

app.py

@@ -6,13 +6,17 @@ import torch
 
 from transformers import DonutProcessor, VisionEncoderDecoderModel
 
+# Check GPU
+device = "cuda" if torch.cuda.is_available() else "cpu"
+    
+# Load processor
+processor = DonutProcessor.from_pretrained("jonathanjordan21/donut_fine_tuning_food_composition_id")
+    
+# Load model
+model = VisionEncoderDecoderModel.from_pretrained("jonathanjordan21/donut_fine_tuning_food_composition_id")
+
 
 def predict(inp):
-    # Load processor
-    processor = DonutProcessor.from_pretrained("jonathanjordan21/donut_fine_tuning_food_composition_id")
-    
-    # Load model
-    model = VisionEncoderDecoderModel.from_pretrained("jonathanjordan21/donut_fine_tuning_food_composition_id")
 
     # Define Json Parser
     def get_komposisi(image_path, image=None):