TokenopolyHQ commited on
Commit
4ae273b
·
1 Parent(s): 8292145

code update in app.py

Browse files
Files changed (2) hide show
  1. app.py +7 -2
  2. requirements.txt +2 -0
app.py CHANGED
@@ -3,6 +3,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
3
  import torch
4
  import json
5
  import re
 
 
6
 
7
  # --------------------------------------------------
8
  # 1. Model loading
@@ -100,7 +102,7 @@ This appears to be a promotional or informational message.
100
  # --------------------------------------------------
101
  # 4. Gradio Blocks UI (Spaces-compatible)
102
  # --------------------------------------------------
103
- with gr.Blocks(api_mode=False) as demo:
104
  gr.Markdown("""
105
  # 🏦 Banking SMS JSON Parser Chatbot
106
 
@@ -132,5 +134,8 @@ with gr.Blocks(api_mode=False) as demo:
132
  # --------------------------------------------------
133
  # 5. App launcher
134
  # --------------------------------------------------
 
 
 
135
  if __name__ == "__main__":
136
- demo.launch()
 
3
  import torch
4
  import json
5
  import re
6
+ from fastapi import FastAPI
7
+ import uvicorn
8
 
9
  # --------------------------------------------------
10
  # 1. Model loading
 
102
  # --------------------------------------------------
103
  # 4. Gradio Blocks UI (Spaces-compatible)
104
  # --------------------------------------------------
105
+ with gr.Blocks() as demo:
106
  gr.Markdown("""
107
  # 🏦 Banking SMS JSON Parser Chatbot
108
 
 
134
  # --------------------------------------------------
135
  # 5. App launcher
136
  # --------------------------------------------------
137
+ app = FastAPI()
138
+ app.mount("/", demo)
139
+
140
  if __name__ == "__main__":
141
+ uvicorn.run(app, host="0.0.0.0", port=7860)
requirements.txt CHANGED
@@ -4,3 +4,5 @@ gradio[oauth]==5.39.0
4
  accelerate==0.24.0
5
  huggingface_hub>=0.33.5
6
  numpy<2
 
 
 
4
  accelerate==0.24.0
5
  huggingface_hub>=0.33.5
6
  numpy<2
7
+ fastapi
8
+ uvicorn