Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import wget
 from transformers import pipeline
 import requests
+import torch
 
 # Nutritionix API setup
 api_url = "https://trackapi.nutritionix.com/v2/natural/nutrients"
@@ -15,11 +16,14 @@ headers = {
 
 # Load the Models
 
+# Check if a GPU is available
+device = 0 if torch.cuda.is_available() else -1
+
 # Load the BLIP VQA Model (Recognize the food)
-visual_quest_ans = pipeline("visual-question-answering", model="Salesforce/blip-vqa-base")
+visual_quest_ans = pipeline("visual-question-answering", model="Salesforce/blip-vqa-base", device=device)
 
 # Load the Translation Model (English to Arabic)
-translation_eng_to_ar = pipeline("translation_en_to_ar", model="marefa-nlp/marefa-mt-en-ar")
+translation_eng_to_ar = pipeline("translation_en_to_ar", model="marefa-nlp/marefa-mt-en-ar", device=device)
 
 # Function to recognize food from the image using the VQA model
 def food_recognizer(image):