s444415/donut
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

show ip and back to orgiinal config

Browse Source
This commit is contained in:
mkozlowskiAzimuthe 2023-01-22 23:50:45 +01:00
parent 28104dd686
commit 8c6786c9cb
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
single_eval.py
View File

@ -6,9 +6,10 @@ import time
from fastapi import FastAPI, UploadFile, File from fastapi import FastAPI, UploadFile, File
import io import io
import os import os
from sys import platform
image_size = [1920, 2560]
print("Set up config") print("Set up config")
image_size = [768, 1280]
config_vision = VisionEncoderDecoderConfig.from_pretrained("Zombely/plwiki-proto-fine-tuned-v3.2") config_vision = VisionEncoderDecoderConfig.from_pretrained("Zombely/plwiki-proto-fine-tuned-v3.2")
config_vision.encoder.image_size = image_size # (height, width) config_vision.encoder.image_size = image_size # (height, width)
config_vision.decoder.max_length = 768 config_vision.decoder.max_length = 768
@ -25,7 +26,10 @@ model.eval()
model.to(device) model.to(device)
print("Print ipconfig") print("Print ipconfig")
os.system("ipconfig") if platform.linux:
os.system("ip r")
else:
os.system("ipconfig")
print("Starting server") print("Starting server")
app = FastAPI() app = FastAPI()