Update app.py

app.py

@@ -6,17 +6,20 @@ import torch
 
 from transformers import DonutProcessor, VisionEncoderDecoderModel
 
-# Check GPU
-device = "cuda" if torch.cuda.is_available() else "cpu"
+def predict(inp):
+
+    # 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 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")
+    # Load model
+    model = VisionEncoderDecoderModel.from_pretrained("jonathanjordan21/donut_fine_tuning_food_composition_id")
+
 
-# Define Json Parser
-def get_komposisi(image_path, image=None):
+    # Define Json Parser
+    def get_komposisi(image_path, image=None):
         image = Image.open(image_path).convert('RGB') if image== None else image.convert('RGB')
     
         task_prompt = "<s_kmpsi>"
@@ -42,8 +45,6 @@ def get_komposisi(image_path, image=None):
     
         return processor.token2json(sequence3)
 
-
-def predict(inp):
     #Generate Output
     out = get_komposisi("", inp)
     return out