import os | |
import getpass | |
from dotenv import load_dotenv | |
load_dotenv() | |
import prompts | |
import gradio_module | |
INTERFACE='GRADIO' | |
MODEL='LLAMA' | |
REPO_ID='meta-llama/Llama-3.2-1B-Instruct' | |
#REPO_ID='meta-llama/Llama-3.2-90B-Vision-Instruct' | |
if INTERFACE=='GRADIO': | |
gradio_module.init_gradio(REPO_ID) | |
gradio_interface=gradio_module.get_gradio_interface() | |
gradio_interface.launch() | |