cafierom commited on
Commit
b715f7c
·
verified ·
1 Parent(s): d9bb849

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +38 -0
app.py ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from typing import Annotated, TypedDict, Literal
3
+ from langchain_community.tools import DuckDuckGoSearchRun
4
+ from langchain_core.tools import tool
5
+ from langgraph.prebuilt import ToolNode, tools_condition
6
+ from langgraph.graph import StateGraph, START, END
7
+ from langgraph.graph.message import add_messages
8
+ from langchain_core.messages import SystemMessage, trim_messages, AIMessage, HumanMessage, ToolCall
9
+
10
+ from langchain_huggingface.llms import HuggingFacePipeline
11
+ from langchain_huggingface import ChatHuggingFace
12
+ from langchain_core.prompts import PromptTemplate, ChatPromptTemplate
13
+ from langchain_core.runnables import chain
14
+ from uuid import uuid4
15
+ import re
16
+ import matplotlib.pyplot as plt
17
+
18
+ from rdkit import Chem
19
+ from rdkit.Chem import AllChem, QED
20
+ from rdkit.Chem import Draw
21
+ from rdkit.Chem.Draw import MolsToGridImage
22
+ from rdkit import rdBase
23
+ from rdkit.Chem import rdMolAlign
24
+ import os, re
25
+ from rdkit import RDConfig
26
+ import pubchempy as pcp
27
+ import gradio as gr
28
+ from PIL import Image
29
+
30
+ device = "cuda" if torch.cuda.is_available() else "cpu"
31
+
32
+ hf = HuggingFacePipeline.from_model_id(
33
+ model_id= "microsoft/Phi-4-mini-instruct",
34
+ task="text-generation",
35
+ pipeline_kwargs = {"max_new_tokens": 500, "temperature": 0.4})
36
+
37
+ chat_model = ChatHuggingFace(llm=hf)
38
+