Sathishkumartheta's picture
Update app.py
f5b8cd0 verified
raw
history blame contribute delete
394 Bytes
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()