sflindrs commited on
Commit
2e39399
·
verified ·
1 Parent(s): 4d90ea3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -22,8 +22,7 @@ model = AutoModelForCausalLM.from_pretrained(
22
  device_map='auto'
23
  )
24
 
25
- import json
26
-
27
  def wrap_json_in_markdown(text):
28
  result = []
29
  stack = []
@@ -184,7 +183,6 @@ def process_text_deepl(input_string, api_key):
184
  processed_string = unicode_word_pattern.sub(process_match, input_string)
185
  return processed_string
186
 
187
- @spaces.GPU()
188
  def process_image_and_text(image, text):
189
  # Process the image and text
190
  inputs = processor.process(
@@ -205,10 +203,10 @@ def process_image_and_text(image, text):
205
  # Only get generated tokens; decode them to text
206
  generated_tokens = output[0, inputs['input_ids'].size(1):]
207
  generated_text = processor.tokenizer.decode(generated_tokens, skip_special_tokens=True)
208
- generated_text_w_json_wrapper = wrap_json_in_markdown(generated_text)
209
- generated_text_w_unicode_mdn = process_text_deepl(generated_text_w_json_wrapper, "a5b1749b-7112-4c2d-81a3-33ea18478bb4:fx")
210
 
211
- return generated_text_w_unicode_mdn
212
 
213
  def chatbot(image, text, history):
214
  if image is None:
 
22
  device_map='auto'
23
  )
24
 
25
+ @spaces.GPU()
 
26
  def wrap_json_in_markdown(text):
27
  result = []
28
  stack = []
 
183
  processed_string = unicode_word_pattern.sub(process_match, input_string)
184
  return processed_string
185
 
 
186
  def process_image_and_text(image, text):
187
  # Process the image and text
188
  inputs = processor.process(
 
203
  # Only get generated tokens; decode them to text
204
  generated_tokens = output[0, inputs['input_ids'].size(1):]
205
  generated_text = processor.tokenizer.decode(generated_tokens, skip_special_tokens=True)
206
+ generated_text_w_unicode_mdn = process_text_deepl(generated_text, "a5b1749b-7112-4c2d-81a3-33ea18478bb4:fx")
207
+ generated_text_w_json_wrapper = wrap_json_in_markdown(generated_text_w_unicode_mdn)
208
 
209
+ return generated_text_w_json_wrapper
210
 
211
  def chatbot(image, text, history):
212
  if image is None: