Update app.py

app.py

@@ -3,12 +3,13 @@ from transformers import AutoProcessor, Blip2ForConditionalGeneration
 import torch
 from PIL import Image
 
-# Set device to GPU if available
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-
 # Load the BLIP-2 model and processor
 processor = AutoProcessor.from_pretrained("Salesforce/blip2-opt-2.7b")
-model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b").to(device)
+model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b")
+
+# Set device to GPU if available
+device = "cuda" if torch.cuda.is_available() else "cpu"
+model.to(device)
 
 def blip2_interface(image, prompted_caption_text, vqa_question, chat_context):
     # Prepare image input
@@ -40,8 +41,8 @@ def blip2_interface(image, prompted_caption_text, vqa_question, chat_context):
 
 # Define Gradio input and output components
 image_input = gr.inputs.Image(type="numpy")
-text_input = gr.components.TextInput()
-output_text = gr.components.Textbox()
+text_input = gr.inputs.Text()
+output_text = gr.outputs.Textbox()
 
 # Create Gradio interface
 iface = gr.Interface(