rafaaa2105 commited on
Commit
a60b778
·
verified ·
1 Parent(s): 0be2d65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -6
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import gradio as gr
 
2
  import requests
3
  from PIL import Image
4
  from transformers import BlipProcessor, BlipForConditionalGeneration
@@ -33,16 +34,17 @@ model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-capt
33
  def caption(img, min_len, max_len):
34
  raw_image = Image.open(img).convert('RGB')
35
 
36
- inputs = processor(raw_image, return_tensors="pt")
37
-
38
- image_out = model.generate(**inputs, min_length=min_len, max_length=max_len)
39
-
40
- text_output = processor.decode(image_out[0], skip_special_tokens=True)
 
41
 
42
  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ê."
43
 
44
 
45
- text_out = generate(text_output, system_prompt, max_len)
46
  return text_out
47
 
48
 
 
1
  import gradio as gr
2
+ from gradio_client import Client
3
  import requests
4
  from PIL import Image
5
  from transformers import BlipProcessor, BlipForConditionalGeneration
 
34
  def caption(img, min_len, max_len):
35
  raw_image = Image.open(img).convert('RGB')
36
 
37
+ client = Client("https://ydshieh-kosmos-2.hf.space/")
38
+ result = client.predict(
39
+ raw_image, # str (filepath or URL to image) in 'Test Image' Image component
40
+ "Detailed", # str in 'Description Type' Radio component
41
+ fn_index=4
42
+ )
43
 
44
  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ê."
45
 
46
 
47
+ text_out = generate(result, system_prompt, max_len)
48
  return text_out
49
 
50