Spaces:
Sleeping
Sleeping
Upload app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ import torch.nn.functional as F
|
|
| 9 |
from flask import Flask, request, jsonify, Response
|
| 10 |
from transformers import AutoTokenizer, AutoModelForMaskedLM, AutoModelForCausalLM, TextIteratorStreamer
|
| 11 |
from threading import Thread
|
| 12 |
-
from flask import send_from_directory
|
| 13 |
|
| 14 |
# 1. Environment Parsing & Architecture Strategy Mapping
|
| 15 |
MODEL_NAME = os.getenv("MODEL_NAME", "dllm-hub/Qwen3-0.6B-diffusion-bd3lm-v0.1")
|
|
|
|
| 9 |
from flask import Flask, request, jsonify, Response
|
| 10 |
from transformers import AutoTokenizer, AutoModelForMaskedLM, AutoModelForCausalLM, TextIteratorStreamer
|
| 11 |
from threading import Thread
|
| 12 |
+
from flask import redirect, send_from_directory
|
| 13 |
|
| 14 |
# 1. Environment Parsing & Architecture Strategy Mapping
|
| 15 |
MODEL_NAME = os.getenv("MODEL_NAME", "dllm-hub/Qwen3-0.6B-diffusion-bd3lm-v0.1")
|