vverma commited on
Commit
7ad0578
·
1 Parent(s): 44c800d

fixed requirements.txt

Browse files
Files changed (2) hide show
  1. app.py +3 -4
  2. 1+7.png → sample.png +0 -0
app.py CHANGED
@@ -7,16 +7,15 @@ app = FastAPI()
7
  @app.get("/")
8
  def greet_json():
9
  # Load model and processor from Hugging Face
 
10
  processor = TrOCRProcessor.from_pretrained('tjoab/latex_finetuned')
11
  model = VisionEncoderDecoderModel.from_pretrained('tjoab/latex_finetuned')
12
 
13
- paths = "./"
14
-
15
  # Load all images as a batch
16
- images = [open_PIL_image(path) for path in paths]
17
 
18
  # Preprocess the images
19
- preproc_image = processor.image_processor(images=images, return_tensors="pt").pixel_values
20
 
21
  # Generate and decode the tokens
22
  # NOTE: max_length default value is very small, which often results in truncated inference if not set
 
7
  @app.get("/")
8
  def greet_json():
9
  # Load model and processor from Hugging Face
10
+ print("Loading model and processor...")
11
  processor = TrOCRProcessor.from_pretrained('tjoab/latex_finetuned')
12
  model = VisionEncoderDecoderModel.from_pretrained('tjoab/latex_finetuned')
13
 
 
 
14
  # Load all images as a batch
15
+ sample_image = open_PIL_image("sample.png")
16
 
17
  # Preprocess the images
18
+ preproc_image = processor.image_processor(images=[sample_image], return_tensors="pt").pixel_values
19
 
20
  # Generate and decode the tokens
21
  # NOTE: max_length default value is very small, which often results in truncated inference if not set
1+7.png → sample.png RENAMED
File without changes