Spaces:

Staticaliza
/

Vision

Running on Zero

Staticaliza commited on Aug 10

Commit

3538bec

•

1 Parent(s): 8a86647

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ footer {
 # Functions
 @spaces.GPU(duration=60)
 def generate(image, instruction=DEFAULT_INPUT, sampling=False, temperature=0.7, top_p=0.8, top_k=100, repetition_penalty=1.05, max_tokens=512):
-    global repo, tokenizer
     print(image)
     # image_rgb = Image.open(image).convert("RGB")

 # Functions
 @spaces.GPU(duration=60)
 def generate(image, instruction=DEFAULT_INPUT, sampling=False, temperature=0.7, top_p=0.8, top_k=100, repetition_penalty=1.05, max_tokens=512):
+    repo.to(DEVICE)
     print(image)
     # image_rgb = Image.open(image).convert("RGB")