Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import os
 import gradio as gr
 import torch
@@ -12,7 +13,7 @@ DEFAULT_PROMPT = "<image>"
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
-model = FlamingoModel.from_pretrained('dhansmair/flamingo-mini')
+model = FlamingoModel.from_pretrained('dhansmair/flamingo-tiny')
 model.to(device)
 model.eval()