Spaces:
Runtime error
Runtime error
rev app
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ import torch
|
|
18 |
torch._dynamo.config.suppress_errors = True
|
19 |
torch._dynamo.config.verbose = False
|
20 |
|
21 |
-
|
22 |
from transformers import TextStreamer
|
23 |
import gradio as gr
|
24 |
from huggingface_hub import login
|
@@ -41,6 +41,7 @@ def load_model():
|
|
41 |
global model, tokenizer
|
42 |
print("กำลังโหลดโมเดล...")
|
43 |
try:
|
|
|
44 |
# โหลด base model และ tokenizer
|
45 |
base_model, tokenizer = FastVisionModel.from_pretrained(
|
46 |
"unsloth/Llama-3.2-11B-Vision-Instruct",
|
|
|
18 |
torch._dynamo.config.suppress_errors = True
|
19 |
torch._dynamo.config.verbose = False
|
20 |
|
21 |
+
|
22 |
from transformers import TextStreamer
|
23 |
import gradio as gr
|
24 |
from huggingface_hub import login
|
|
|
41 |
global model, tokenizer
|
42 |
print("กำลังโหลดโมเดล...")
|
43 |
try:
|
44 |
+
from unsloth import FastVisionModel
|
45 |
# โหลด base model และ tokenizer
|
46 |
base_model, tokenizer = FastVisionModel.from_pretrained(
|
47 |
"unsloth/Llama-3.2-11B-Vision-Instruct",
|