Hjgugugjhuhjggg commited on
Commit
d27b777
·
verified ·
1 Parent(s): eda5cd9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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,