Tulitula commited on
Commit
2abdc87
·
verified ·
1 Parent(s): 690d42b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -1,8 +1,5 @@
1
- # app.py
2
-
3
  import re
4
  import gradio as gr
5
- from PIL import Image
6
  from transformers import pipeline
7
 
8
  # 1) Image-to-text: ChatDOC/OCRFlux-3B for rich description
@@ -11,7 +8,7 @@ image_to_text = pipeline(
11
  model="ChatDOC/OCRFlux-3B"
12
  )
13
 
14
- # 2) Helper to build Flan-T5-small text pipelines (temp=1.0)
15
  def make_pipeline(model_name, max_tokens):
16
  return pipeline(
17
  "text2text-generation",
 
 
 
1
  import re
2
  import gradio as gr
 
3
  from transformers import pipeline
4
 
5
  # 1) Image-to-text: ChatDOC/OCRFlux-3B for rich description
 
8
  model="ChatDOC/OCRFlux-3B"
9
  )
10
 
11
+ # 2) Helper to build Flan-T5-small text pipelines (temperature=1.0)
12
  def make_pipeline(model_name, max_tokens):
13
  return pipeline(
14
  "text2text-generation",