Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -4,6 +4,8 @@ from fastapi import FastAPI, HTTPException
|
|
4 |
from fastapi.responses import StreamingResponse
|
5 |
from pydantic import BaseModel, field_validator
|
6 |
from transformers import (
|
|
|
|
|
7 |
AutoModelForCausalLM,
|
8 |
AutoTokenizer,
|
9 |
GenerationConfig,
|
|
|
4 |
from fastapi.responses import StreamingResponse
|
5 |
from pydantic import BaseModel, field_validator
|
6 |
from transformers import (
|
7 |
+
AutoConfig,
|
8 |
+
pipeline,
|
9 |
AutoModelForCausalLM,
|
10 |
AutoTokenizer,
|
11 |
GenerationConfig,
|