app.py

# -*- coding: utf-8 -*-
"""app.ipynb

Automatically generated by Colab.

Original file is located at
    https://colab.research.google.com/drive/1ewSnt_PQx8cX9GC2XAguVNesbn8MM3tA
"""



from fastapi import FastAPI
from pydantic import BaseModel
from transformers import RobertaTokenizer, RobertaForMaskedLM, pipeline

app = FastAPI()


model_name = "web3se/SmartBERT-v3"
tokenizer = RobertaTokenizer.from_pretrained(model_name)
model = RobertaForMaskedLM.from_pretrained(model_name)

# Create fill-mask pipeline
fill_mask = pipeline('fill-mask', model=model, tokenizer=tokenizer)

# Define request body
class ContractRequest(BaseModel):
    contract_code: str

@app.post("/analyze/")
async def analyze_contract(request: ContractRequest):
    # Use fill-mask on contract code
    outputs = fill_mask(request.contract_code)

    return {"predictions": outputs}  # Returns top masked token predictions


# Run API on Hugging Face Spaces
if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=7860)