DocterXray-v1.2

Running on Zero

Aekanun commited on Nov 26, 2024

Commit

b3e3425

1 Parent(s): 7c418c1

rev app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -158,7 +158,7 @@ def load_model():
         traceback.print_exc()
         return False
-@spaces.GPU(duration=90)
 def process_image(image):
     # # ย้ายมาไว้ตรงนี้
     # import os
@@ -227,7 +227,7 @@ def process_image(image):
         outputs = model.generate(
             **inputs,
             streamer=text_streamer,
-            max_new_tokens=128,
             use_cache=True,
             temperature=1.5,
             min_p=0.1

         traceback.print_exc()
         return False
+@spaces.GPU(duration=120)
 def process_image(image):
     # # ย้ายมาไว้ตรงนี้
     # import os
         outputs = model.generate(
             **inputs,
             streamer=text_streamer,
+            max_new_tokens=256,
             use_cache=True,
             temperature=1.5,
             min_p=0.1