MekkCyber commited on
Commit
637d627
·
1 Parent(s): d51526b
Files changed (2) hide show
  1. app.py +9 -9
  2. requirements.txt +2 -1
app.py CHANGED
@@ -5,7 +5,10 @@ import time
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
  import logging
7
  from starlette.middleware.sessions import SessionMiddleware
8
-
 
 
 
9
 
10
  # Configure logging
11
  logging.basicConfig(level=logging.INFO)
@@ -137,12 +140,9 @@ def interface():
137
  return demo
138
 
139
  demo = interface()
140
-
141
- # Access FastAPI app instance from Gradio
142
- fastapi_app = demo.app
143
-
144
- # Add SessionMiddleware to enable session management
145
- fastapi_app.add_middleware(SessionMiddleware, secret_key="secret_key") # Use a secure, random secret key
146
-
147
  # Launch the app
148
- demo.launch()
 
 
 
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
  import logging
7
  from starlette.middleware.sessions import SessionMiddleware
8
+ from fastapi import FastAPI
9
+ import uvicorn
10
+ app = FastAPI()
11
+ app.add_middleware(SessionMiddleware, secret_key="secure_key")
12
 
13
  # Configure logging
14
  logging.basicConfig(level=logging.INFO)
 
140
  return demo
141
 
142
  demo = interface()
143
+ app.mount("/", demo)
144
+
 
 
 
 
 
145
  # Launch the app
146
+ # demo.launch()
147
+
148
+ uvicorn.run(app, host="0.0.0.0", port=7860)
requirements.txt CHANGED
@@ -5,4 +5,5 @@ huggingface-hub
5
  transformers
6
  torch
7
  gradio[oauth]
8
- starlette
 
 
5
  transformers
6
  torch
7
  gradio[oauth]
8
+ starlette
9
+ uvicorn