aexyb's picture
Update app.py
1d2c609 verified
raw
history blame
No virus
621 Bytes
import gradio as gr
import concurrent.futures
# Load the model into RAM
model = gr.load("models/TheBloke/SOLAR-10.7B-Instruct-v1.0-uncensored-GGUF")
def interact(input):
# Define the function for user interaction
response = model(input)
return response
# Use ThreadPoolExecutor to manage the threads
with concurrent.futures.ThreadPoolExecutor(max_workers=2) as executor:
# Create a Gradio interface with the loaded model
interface = gr.Interface(fn=interact, inputs="text", outputs="text")
# Handle the interactions with Gradio
interface.launch()