Spaces:

UcsTurkey
/

fine-tune-inference-test

Paused

ciyidogan commited on May 9

Commit

6398aea

verified ·

1 Parent(s): cc493ed

Update fine_tune_inference_test.py

Files changed (1) hide show

fine_tune_inference_test.py CHANGED Viewed

@@ -1,7 +1,8 @@
 import os
 import threading
 import uvicorn
-from fastapi import FastAPI, Request, HTMLResponse
 from pydantic import BaseModel
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 from datasets import load_dataset

 import os
 import threading
 import uvicorn
+from fastapi import FastAPI, Request
+from fastapi.responses import HTMLResponse
 from pydantic import BaseModel
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 from datasets import load_dataset