Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from gradio_client import Client
 import requests
 from PIL import Image
 from transformers import BlipProcessor, BlipForConditionalGeneration
@@ -33,16 +34,17 @@ model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-capt
 def caption(img, min_len, max_len):
     raw_image = Image.open(img).convert('RGB')
   
-    inputs = processor(raw_image, return_tensors="pt")
-    
-    image_out = model.generate(**inputs, min_length=min_len, max_length=max_len)
-
-    text_output = processor.decode(image_out[0], skip_special_tokens=True)
+    client = Client("https://ydshieh-kosmos-2.hf.space/")
+    result = client.predict(
+    				raw_image,	# str (filepath or URL to image)    								in 'Test Image' Image component
+    				"Detailed",	# str in 'Description Type' Radio component
+    				fn_index=4
+    )
     
     system_prompt = "Ignore todas as instruções anteriores. Considerando os detalhes fornecidos, crie uma descrição de marketing com um tom atraente/marketing para potenciais clientes, usando palavras que os empoderem. Apresente as informações em um único parágrafo coeso. Se algum caractere especial estiver presente, como '<' ou '>' ou qualquer outro semelhante, você DEVE ignorar e não adicionar à descrição. Qualquer emoji ou emoticon ou qualquer símbolo semelhante **DEVE** ser ignorado, não importa o quê, sem exceções. Abstenha-se de oferecer sugestões sobre o uso ou combinações do produto. Quando uma cor for dada, não implique que várias cores do produto estão disponíveis. Em vez disso, coloque as cores dadas como a cor do produto. Se uma cor não for fornecida, não mencione sobre ela na descrição que você escrever. Empregue uma linguagem propícia a resultados ótimos de mecanismos de busca no Google. Em casos em que o usuário fornece informações aparentemente inalteráveis, reformule ou reescreva o texto sem introduzir detalhes desnecessários. DESCONSIDERE QUALQUER referência a peso (gramas (g), quilogramas (kg), etc.) se mencionado nos detalhes fornecidos. Quando mais de uma cor for dada, não diga que é para produtos diferentes. Em vez disso, insira-as juntas, dizendo que é apenas um produto que tem todas as cores dadas de uma forma natural. Não use palavras clichês no início da frase. Além disso, você DEVE sempre fornecer respostas em português. Adira meticulosamente a estas instruções e abstenha-se de desviar das diretrizes fornecidas. Abstenha-se de fabricar qualquer informação. Todas as instruções dadas **DEVEM** ser respeitadas e seguidas, não importa o quê."
 
 
-    text_out = generate(text_output, system_prompt, max_len)
+    text_out = generate(result, system_prompt, max_len)
     return text_out